jax>=0.4.20
jaxlib>=0.4.20
flax>=0.7.5
numpy>=1.24

[reference]
ablang2
torch
