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

[linear]
grgrlib>=0.1.22
