train.py is still missing and I couldn't find the solution in the closed issue.

Hi Ricky, are we supposed to run it via levers.py? Since it has the train function.
Also, which version of PyTorch has this code been written it?