jax>=0.4.13
jaxtyping>=0.2.20
typing_extensions>=4.5.0
equinox>=0.11.3
optax>=0.2.0
tqdm>=4.63.2
matplotlib>=3.8.1
pandas>=2.2.0
seaborn>=0.13.0
exponax==0.1.0
pdequinox==0.1.2
trainax==0.0.2
