dm_haiku==0.0.9
e3nn_jax==0.12.0
jax==0.3.25
jraph==0.0.6.dev0
numpy>=1.23.4
optax==0.1.3
