-
Notifications
You must be signed in to change notification settings - Fork 57
Pull requests: jax-ml/jax-triton
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
0.5.0 add support for most Triton's features; large rewrite/refactoring
#383
opened Apr 16, 2026 by
Arech8
Contributor
Loading…
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…
Fix: Add oryx to optional deps and patch ROCm resource leak
#347
opened Jun 26, 2025 by
Abinesh-Mathivanan
Loading…
[JAX] Wrap triton_call custom call using the FFI.
#345
opened Jun 3, 2025 by
copybara-service
Bot
Loading…
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…
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…
Raise exception on too much shared memory requested
#160
opened Jun 5, 2023 by
copybara-service
Bot
Loading…
Fix dot to use preferred_element_type
pull ready
#39
opened Nov 16, 2022 by
sharadmv
Collaborator
Loading…
ProTip!
Exclude everything labeled
bug with -label:bug.