numpy<2.0
torch>=2.3
packaging
tqdm
pyyaml
transformers
datasets
protobuf
SentencePiece
rich
omegaconf
pydantic
click
click_pathlib
jq
class_resolver
wandb
einops>=0.7.0
flash-attn

[install_helper]
ninja

[linting]
pre-commit

[tests]
pytest
pytest-cov
