jax

[dev]
ruff
pyright
pytest
furo
sphinx-autoapi
