torch
datasets==3.6.0
einops
fsspec
git-lfs
h5py
hydra-core
ipdb
lightning
notebook
nvitop
omegaconf
pandas
rich
seaborn
scikit-learn
timm
transformers
wandb
packaging
mauve-text
setproctitle
mamba-ssm
flash-attn
causal-conv1d