jax>=0.10.0
jaxlib>=0.10.0
flax>=0.12.8
numpy>=2.0

[all]
jraphx[dev,docs,examples,test]

[dev]
pytest>=7.0
pytest-xdist
black>=22.0
isort>=5.10
mypy>=0.990
ruff>=0.1.0
pre-commit>=3.0

[docs]
sphinx>=7.4
sphinx-autodoc-typehints>=2.3
sphinx-copybutton>=0.5.2
sphinx-rtd-theme>=3.0

[examples]
optax>=0.2.5
torch-geometric>=2.0.0
grain>=0.2.12

[test]
pytest>=7.0
pytest-xdist
optax>=0.2.5
