jax>=0.4.38
optax
jaxtyping

[all]
hyperoptax[dev,docs,notebooks,testing]

[dev]
ruff
mypy
pre-commit

[notebooks]
flax
jupyter
matplotlib
notebook
rejax>=0.1.2
tqdm
bayesian-optimization
bayex

[testing]
pytest>=8.4.1
pytest-cov>=6.2.1
