jax==0.6.0

[cuda12]
jax[cuda12]==0.6.0

[dev]
ruff
pytest
pytest-cov
coverage-badge
sphinx-lint>=1.0.0
typos
