pybind11==2.13.6
numpy==1.26.4
datasets==3.2.0
dill==0.3.8
transformers==4.49.0
peft==0.14.0
pylatexenc==2.10
omegaconf==2.3.0
sentence_transformers==3.4.1
tabulate==0.9.0
orjson==3.10.15
codetiming==1.4.0
google-cloud-aiplatform==1.82.0
antlr4-python3-runtime==4.9.0
grpcio==1.71.0rc2
scikit-learn==1.5.1
hydra-core==1.3.2
wandb==0.19.2
pyarrow==18.1.0
ray[data,train,tune,serve]==2.40.0
tensordict==0.5.0
pynvml==12.0.0
flash-attn==2.7.3