Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Clarify that jax-triton only works on GPU #276

Open
skye opened this issue Apr 18, 2024 · 0 comments
Open

Clarify that jax-triton only works on GPU #276

skye opened this issue Apr 18, 2024 · 0 comments

Comments

@skye
Copy link

skye commented Apr 18, 2024

There's been some confusion as to whether this package includes TPU support for Triton. If anyone finds this issue -- it doesn't! This is just integration JAX with existing GPU-only Triton. We should consider briefly clarify this in the README.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant