torch.py

pip install torch