# 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