jax>=0.4.30
jaxlib>=0.4.30
numpy
pyyaml

[all]
pallas-forge[dev,profile,viz]

[build]
build>=1.0
twine>=5.0

[dev]
pytest>=7.0
ruff>=0.4.0
pre-commit>=3.7

[profile]
tensorboard-plugin-profile

[tpu]
jax[tpu]>=0.4.30

[viz]
matplotlib>=3.7
seaborn>=0.12
