Skip to content

Pull requests: jax-ml/jax-triton

New pull request New
19 Open 245 Closed
19 Open 245 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add Heuristics and Autotuner to type annotation
#323 opened Dec 22, 2024 by stephen-huan Loading…
Add CPU backend
#322 opened Dec 21, 2024 by stephen-huan • Draft
Fix tests
#321 opened Dec 21, 2024 by stephen-huan Loading…
Update references to JAX's GitHub repo
#316 opened Dec 11, 2024 by copybara-service bot Loading…
Replace Deprecated abc Decorators (#1)
#314 opened Nov 8, 2024 by pixeeai Loading…
2
Add if conditional to GPU tests to only run on main repo pull ready
#312 opened Oct 25, 2024 by MichaelHudgins Loading…
Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a)
#300 opened Oct 11, 2024 by copybara-service bot Loading…
Check that a kernel can be launched before adding it to candidate kernels
#248 opened Jan 9, 2024 by copybara-service bot Loading…
1
More flexible debugging of triton IRs
#223 opened Aug 23, 2023 by copybara-service bot Loading…
1
Add segment_ids support to pallas flash attention on GPU.
#193 opened Jul 12, 2023 by copybara-service bot Loading…
A faster flash attention bwd implementation
#177 opened Jun 22, 2023 by tonywu95 Loading…
30
Raise exception on too much shared memory requested
#160 opened Jun 5, 2023 by copybara-service bot Loading…
1
Add a Softmax Triton implementation.
#154 opened May 25, 2023 by copybara-service bot Loading…
Pallas docs
#146 opened May 22, 2023 by sharadmv • Draft
Update Triton kernel example
#134 opened May 6, 2023 by wangkuiyi Loading…
2
Pallas autotuning
#108 opened Mar 30, 2023 by sharadmv Loading…
Remove workaround for triton compiler bug.
#92 opened Feb 23, 2023 by copybara-service bot Loading…
1
WIP AD of pallas
#85 opened Feb 14, 2023 by sharadmv • Draft
Fix dot to use preferred_element_type pull ready
#39 opened Nov 16, 2022 by sharadmv Loading…
ProTip! What’s not been updated in a month: updated:<2024-12-07.