TemporalMesh-Transformer / requirements.txt
vigneshwar234's picture
Add requirements.txt
836634c verified
raw
history blame contribute delete
335 Bytes
# Core
torch>=2.2.0
torchvision
torchaudio
# TMT-specific
einops>=0.7.0
torch-geometric
# NLP / data
transformers>=4.40.0
datasets>=2.19.0
tokenizers>=0.19.0
# Experiment tooling
wandb>=0.17.0
tensorboard>=2.16.0
torchviz
matplotlib>=3.8.0
jupyter>=1.0.0
pandas>=2.2.0
tabulate>=0.9.0
# Dev
pytest>=8.0.0
black>=24.0.0
ruff>=0.4.0