jax>=0.3.16
numpy>=1.22.4

[dev]
pytest
pytest-cov
coverage[toml]
ruff
