torch>=2.0.0
torchvision>=0.15.1
torchaudio>=2.0.1
nnsight>=0.1.0
transformers>=4.30.0
datasets>=2.0.0
wandb>=0.15.0
tqdm>=4.64.0
numpy>=1.24.0
zstandard
h5py
requests

[dev]
pytest
pytest-cov
black>=23.0.0
flake8>=6.0.0
mypy>=1.0.0
