absl-py>=1.4.0
jax>=0.4.34
triton>=3.1

[tests]
pytest
