apex==0.9.10dev
bfloat16==1.2.0
causal_conv1d==1.5.0.post8
cupy==13.4.1
einops==0.8.1
flash_attn==2.7.1.post4
fvcore==0.1.5.post20221221
hydra-core==1.3.2
lightning_fabric==2.5.1
lightning_lite==1.8.6
mamba_ssm==2.2.4
matplotlib==3.10.1
nltk==3.9.1
numpy==2.2.4
nvidia_ml_py==12.560.30
omegaconf==2.3.0
packaging==24.2
pandas==2.2.3
Pillow==11.1.0
psutil==6.1.0
pydot==3.0.4
pytest==8.3.4
python-dotenv==1.1.0
pytorch_lightning==2.5.1
rich==14.0.0
rmm==0.0.1
safetensors==0.5.3
seaborn==0.13.2
sentencepiece==0.2.0
setuptools==75.6.0
tabulate==0.9.0
torch==2.5.1
torchmetrics==1.7.1
torchvision==0.20.1
tqdm==4.67.1
transformers==4.50.1
wheel==0.45.1
xformers==0.0.28.post3
