distrax==0.1.5
flax==0.8.0
optax==0.1.9
orbax-checkpoint==0.5.3
tensorboardX==2.6.2.2
tqdm==4.66.1
numpy==1.26.4
tyro==0.7.1
pandas==2.2.0
seaborn==0.13.2
chex==0.1.85
brax==0.10.4
packaging==24.0
einops==0.8.0
huggingface-hub==0.23.0
safetensors==0.4.3
hydra-core==1.3.2
ninja==1.11.1.1