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

[linear]
grgrlib>=0.1.22
