Skip to content

Pull requests: jax-ml/jax-triton

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
Assignee
Filter by who’s assigned
Sort

Pull requests list

Replace Deprecated abc Decorators (#1)
#314 opened Nov 8, 2024 by pixeeai Loading…
More flexible debugging of triton IRs
#223 opened Aug 23, 2023 by copybara-service bot Loading…
A faster flash attention bwd implementation
#177 opened Jun 22, 2023 by tonywu95 Loading…
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…
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…
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! Adding no:label will show everything without a label.