# torch
torch==2.6.0
torchvision
torchaudio
--extra-index-url https://download.pytorch.org/whl/cu124

# base
accelerate
cryptography
deepspeed
hydra-core
matplotlib
PyYAML
pybind11==2.13.6
sentencepiece
tensorboard
transformers

# test
coverage
pytest
pytest_asyncio
pytest-cov
pytest_mock
pytest-random-order
wandb