pandas~=2.2
jax==0.4.27
jaxlib==0.4.26
jaxtyping==0.2.28
chex==0.1.86
optax==0.2.2
tqdm~=4.66
matplotlib~=3.8
pdoc~=14.4
ipywidgets~=8.1
black~=24.4
jupyter-black~=0.3
pytest~=8.2
trajax @ git+XXXX
wandb
