absl-py>=1.4.0
jax>=0.11.0
triton>=3.7

[tests]
pytest
