jax
jaxlib
tqdm
matplotlib
diffrax
optax
jaxopt
optimistix
scipy
jaxkd
netcdf4
f90nml
pyevtk
optuna
pandas
