Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
144 Open 1,626 Closed
144 Open 1,626 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

Try new collective permute cse
#2324 opened Mar 25, 2026 by wsmoses Loading…
1
Try recognize reduce window
#2323 opened Mar 25, 2026 by wsmoses Loading…
2
Fix BroadcastIotaSimplify using wrong iota dimension for arithmetic sequence broadcasts
#2322 opened Mar 25, 2026 by Copilot AI • Draft
@avik-pal
add a patch to bump stablehlo to https://github.com/openxla/stablehlo/commit/b744df4fa70fa87ccb04b3809dffcded294d8c33
#2318 opened Mar 25, 2026 by Pangoraw Loading…
Display kernel stats and framework op stats in GB-25 CI
#2316 opened Mar 24, 2026 by gbaraldi Loading…
Add 'multi_buffer' to reactant_commit in workflow
#2311 opened Mar 24, 2026 by wsmoses Loading…
add pad for alginment pass
#2307 opened Mar 23, 2026 by wsmoses Loading…
Add 'divisible-grid' to gb25_commit in workflow
#2306 opened Mar 23, 2026 by wsmoses Loading…
7
Add comm pipelining
#2296 opened Mar 21, 2026 by wsmoses Loading…
2
support llvm.alloca for ParallelizeBlockOps and fix while2for conversion
#2295 opened Mar 21, 2026 by xys-syx Loading…
5
[CI] Disable more HLO passes in GB-25 test
#2294 opened Mar 21, 2026 by giordano Loading…
3
Try to do collective combining before compute fusion
#2291 opened Mar 21, 2026 by wsmoses Loading…
Fix XLA concat cost model penalty for stencil codes
#2281 opened Mar 20, 2026 by gbaraldi Loading…
1
Add XLA patch: occupancy-aware fusion cost model
#2277 opened Mar 20, 2026 by gbaraldi Loading…
2 of 3 tasks
Add XLA patch: collective combiner skip reachable instead of breaking
#2273 opened Mar 19, 2026 by gbaraldi Loading…
3 tasks done
6
chore: bump xprof version
#2271 opened Mar 19, 2026 by avik-pal Loading…
13
Try bf16
#2269 opened Mar 18, 2026 by wsmoses Loading…
1
redirect slice through slice
#2268 opened Mar 18, 2026 by mofeing Loading…
5
simplify distributive multiply + add
#2261 opened Mar 17, 2026 by mofeing • Draft
Try improving the gpu spill cost model
#2258 opened Mar 16, 2026 by wsmoses Loading…
1
Fix formatting in test-gb-25 workflow file
#2255 opened Mar 16, 2026 by wsmoses Loading…
GPU: try to improve fusion capabilities
#2246 opened Mar 15, 2026 by wsmoses Loading…
[gb-25] Experiment with different grid sizes
#2234 opened Mar 14, 2026 by giordano Loading…
4
reduce bcast of dot_general contract dims
#2216 opened Mar 12, 2026 by mofeing Loading…
WIP ProbProg
#2213 opened Mar 12, 2026 by sbrantq • Draft
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.