jax>=0.9.2
jaxlib>=0.9.2
flax>=0.12.4
chex>=0.1.91
gymnasium>=1.2.3

[cuda]
jax[cuda]>=0.9.2

[cuda12]
jax[cuda12]>=0.9.2

[tests]
numpy>=2.3.5
pytest>=9.0.2
pytest-xdist>=3.8.0
pytest-html>=4.2.0

[tpu]
jax[tpu]>=0.9.2

[trainer]
optax>=0.2.6
click>=8.3.1
colorama>=0.4.6
tqdm>=4.67.3
pyyaml>=6.0.3
pandas>=3.0.1
fastparquet>=2025.12.0
pyarrow>=23.0.1
duckdb>=1.4.4
rich>=14.3.3
