Update requirements

requirements.txt

@@ -6,14 +6,15 @@ matplotlib
 tqdm
 rich
 torchtext
+lit # Getting installation errors with torch 2.0 if this isn't installed
 # torchvision
 pytorch-lightning==1.9.3
 hydra-core
 omegaconf
 wandb
 einops
 cmake # For pykeops support
-pykeops # If there are installation problems with pykeops==2.x, try pykeops==1.5
+# pykeops # Seems to cause various issues; leaving uninstalled by default
 transformers # For some schedulers
 
 # Model specific packges