torch==2.2.0
transformers==4.42.4
accelerate==0.30.1
datasets==2.16.1
rich==13.7.1
scikit-learn==1.4.2
tiktoken==0.7.0
pytest==8.2.2
bitsandbytes==0.43.3
lightning==2.2.5
hydra-core==1.3.2
hydra_colorlog==1.2.0
wandb==0.16.6
art==6.1
pprintpp==0.4.0
colorama==0.4.6
jsonlines==4.0.0
loguru==0.7.2
goldenretriever-core==0.9.4
setuptools==75.1.0
twine==5.1.1
flash-attn==2.5.9.post1
