jax[cuda]>=0.3.15
triton>=2.0.0.dev20220720
