Skip to content

Pull requests: jax-ml/jax-triton

New pull request New
25 Open 294 Closed
25 Open 294 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Pass triton's global scratch buffer size through to kernel launch
#390 opened Jul 2, 2026 by pashu-cohere Loading…
1
Support extracting custom call cost estimates for JAX Triton kernels
#389 opened Jun 24, 2026 by copybara-service Bot Loading…
0.5.0: finalize updates, drop aliased-input support from zeroed_outputs
#388 opened May 19, 2026 by Arech8 Contributor Loading…
@superbobry
Add @jt.kernel decorator for Triton-style kernel[grid](...) launches
#387 opened May 19, 2026 by Arech8 Contributor Loading…
@superbobry
Refactor triton_call to use Triton binder and support kwargs/tuples
#386 opened May 19, 2026 by Arech8 Contributor Loading…
@superbobry
87
Add TMA support: TensorDescriptor API, scratch metadata, and v2 handl…
#384 opened May 7, 2026 by christopherbate • Draft
1
0.5.0 add support for most Triton's features; large rewrite/refactoring
#383 opened Apr 16, 2026 by Arech8 Contributor Loading…
@superbobry
46
Integrating changes from https://github.com/jax-ml/jax-triton/pull/379
#381 opened Mar 13, 2026 by copybara-service Bot Loading…
Integrate Triton up to [ba7386d](https://github.com/openai/triton/commits/ba7386d942252fcbb336c20cfe5c92d5ee64774f)
#366 opened Nov 19, 2025 by copybara-service Bot Loading…
[JAX] Wrap triton_call custom call using the FFI.
#345 opened Jun 3, 2025 by copybara-service Bot Loading…
Add CPU backend
#322 opened Dec 21, 2024 by stephen-huan Contributor • Draft
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 Contributor 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 Contributor Loading…
30
Raise exception on too much shared memory requested
#160 opened Jun 5, 2023 by copybara-service Bot Loading…
1
Pallas docs
#146 opened May 22, 2023 by sharadmv Collaborator • Draft
Update Triton kernel example
#134 opened May 6, 2023 by wangkuiyi Contributor Loading…
2
Pallas autotuning
#108 opened Mar 30, 2023 by sharadmv Collaborator 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 Collaborator • Draft
Fix dot to use preferred_element_type pull ready
#39 opened Nov 16, 2022 by sharadmv Collaborator Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.