jax>=0.3.20
jaxlib
grgrjax
pyyaml

[linear]
grgrlib>=0.1.22
