Skip to content

Issues: jax-ml/jax-triton

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.

Sign up for GitHub

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

36 Open 18 Closed
36 Open 18 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Clarify that jax-triton only works on GPU
#276 opened Apr 18, 2024 by skye
Is jax-triton replaced by Pallas?
#273 opened Apr 11, 2024 by gautierronan
2
cannot import name 'ShapedArrayRef' due to deprecation in jax._src.state
#271 opened Mar 12, 2024 by dtunai
Unable to install Jax-Triton
#266 opened Feb 27, 2024 by bsaoptima
1
Import error encountered in jax_triton
#264 opened Feb 22, 2024 by egg5154
7
Changes required for AMD/rocm support with Triton
#257 opened Jan 22, 2024 by reachtarunhere
Unable to Install
#246 opened Jan 3, 2024 by karan-dalal
2
Problems in Installing at HEAD
#244 opened Jan 2, 2024 by LeoXinhaoLee
2
triton requirement is out of date
#238 opened Oct 29, 2023 by ywrt
1
Pallas Operators
#228 opened Sep 2, 2023 by srush
XlaRuntimeError when applying FusedAttention
#225 opened Aug 28, 2023 by luise1030
Running Pallas
#224 opened Aug 23, 2023 by mehdiataei
1
Pallas upstream is now working?
#219 opened Aug 12, 2023 by sh0416
2
Hopper Support?
#215 opened Aug 2, 2023 by IzzyPutterman
5
triton dtype mapping for float inputs
#210 opened Jul 27, 2023 by karen-sy
1
ptxas unsupported version: Is CUDA 11.8 support broken in HEAD install?
#208 opened Jul 27, 2023 by markschoene
2
Feature request: lower jnp.exp2 to tl.math.exp2 in pallas
#204 opened Jul 18, 2023 by hr0nix
@mattjj
10
Is scan supported in pallas?
#202 opened Jul 17, 2023 by hr0nix
5
Why does pl.dot explicitly convert its output to float32?
#199 opened Jul 16, 2023 by hr0nix
1
Kernel compilation hangs with a particular dtype
#197 opened Jul 16, 2023 by hr0nix
1
Inconsistent NaN results on Triton matmul kernel
#185 opened Jun 28, 2023 by karen-sy
1
Jax-triton requires python >= 3.10?
#184 opened Jun 27, 2023 by tgunter
1
Example notebook with attention isn't working
#180 opened Jun 24, 2023 by hr0nix
3
Pallas Broken after making JAX-Triton calls serializable update
#179 opened Jun 24, 2023 by adam-hartshorne
6
Incorrect dtype cast in Matmul Example?
#174 opened Jun 17, 2023 by adam-hartshorne
ProTip! Mix and match filters to narrow down what you’re looking for.