thrml>=0.1.0
torch>=2.4
jax>=0.4.30
jaxlib>=0.4.30
equinox>=0.11.4
jaxtyping>=0.2.31
torchax>=0.0.9

[dev]
pytest>=8.0
pytest-xdist>=3.5
rich>=13.7
ruff>=0.6
pre-commit>=3.7

[torchax]
torchax>=0.0.10
