jax<0.5.1,>=0.4.13
jaxlib<0.5.1,>=0.4.13
grgrjax>=0.4.7
pyyaml
scipy

[linear]
grgrlib>=0.1.22
