ase
e3x
flax
ipi
jax
jax-pme>=0.1
jaxtyping
marathon-train[grain]
grain<0.2.17
opsis
optax
