jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
354 stars 40 forks source link

Update jax-triton metadata to depend on the Triton 3.0.0 release. #283

Closed hawkinsp closed 3 months ago

hawkinsp commented 3 months ago

This seems to work just fine! And it will make the pip install considerably easier.

hawkinsp commented 3 months ago

Tests pass with the triton 3.0 release in a VM.