jax>=0.4.13
jaxlib>=0.4.13
grgrjax>=0.4.6
pyyaml
scipy

[linear]
grgrlib>=0.1.22
