You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi there. I'm attempting to follow the installation directions. First, I install jax triton using pip install jax-triton.
Then, I install JAX and JAXLib using the Install JAX Docs: pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Currently, jax-triton installs JAX version 0.4.14. But there are no compatible JAXlib versions available. Thus, I install the most up to date JAX / JAXlib using the command above.
I keep getting the following error message when running the example code: ValueError: triton_call is only available when triton is installed.
What is the correct / updated way to install JAX Triton? It would be great to have some clearer docs, I'm happy to make a PR once I can get it working.
The text was updated successfully, but these errors were encountered:
Hi there. I'm attempting to follow the installation directions. First, I install jax triton using
pip install jax-triton
.Then, I install JAX and JAXLib using the Install JAX Docs:
pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Currently, jax-triton installs JAX version 0.4.14. But there are no compatible JAXlib versions available. Thus, I install the most up to date JAX / JAXlib using the command above.
I keep getting the following error message when running the example code:
ValueError: triton_call is only available when triton is installed.
What is the correct / updated way to install JAX Triton? It would be great to have some clearer docs, I'm happy to make a PR once I can get it working.
The text was updated successfully, but these errors were encountered: