jax<0.4.6,>=0.3.20
jaxlib<0.4.6,>=0.3.20
grgrjax>=0.4.1
pyyaml
scipy

[linear]
grgrlib>=0.1.22
