Simplify dependencies and remove explicit PyTorch requirements

requirements.txt

@@ -118,12 +118,12 @@ tiktoken==0.8.0
 timeout-decorator==0.5.0
 tokenizers==0.19.1
 tomli==2.2.1
-torch==2.0.1
-torchaudio==2.0.2
-torchvision==0.15.2
+torch
+torchaudio
+torchvision
 tqdm==4.67.1
 transformers==4.41.1
-triton==2.1.0
+triton
 typing-inspect==0.9.0
 watchdog==4.0.2
 wrapt==1.17.0