typing-extensions>=4.12

[dev]
pytest>=8.0

[jax]
jax>=0.4.38
optax>=0.2.4

[perf]
psutil>=7.0
