jax
plum-dispatch

[dev]
ruff
jupyterlab
pre-commit
pytest
tox-uv
pyright

[test]
pytest
tox-uv
pytest_cases
