Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
166 Open 1,727 Closed
166 Open 1,727 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

Add PDLL implementation of SliceExtend
#2459 opened May 2, 2026 by jumerckx Collaborator • Draft
4
Refactor to remove nanobind depdency outside of python build
#2455 opened Apr 29, 2026 by Antipath1 Collaborator Loading…
1
add WhileCyclicTranposePropagate transform pattern
#2450 opened Apr 27, 2026 by mofeing Collaborator Loading…
4
Cublas Support
#2444 opened Apr 22, 2026 by dsding2 • Draft
1
refactor: update ScatterToDynamicUpdateSlice to detect special scatter update detection
#2442 opened Apr 21, 2026 by avik-pal Collaborator Loading…
5
feat: extend scatter to dynamic update slice simplification
#2439 opened Apr 19, 2026 by avik-pal Collaborator • Draft
1
Tessera Pattern Matching with Struct Arguments
#2437 opened Apr 17, 2026 by jessicacotturone21 Collaborator • Draft
4
add diff rules for stablehlo.triangular_solve
#2436 opened Apr 16, 2026 by mofeing Collaborator • Draft
2
test complex diff of rules which use HLOConstantFP
#2435 opened Apr 16, 2026 by mofeing Collaborator • Draft
Add outline-enzyme-regions into raise pipeline
#2434 opened Apr 15, 2026 by xys-syx Collaborator Loading…
2
TridiagonalSolveOp Lowering
#2433 opened Apr 15, 2026 by snonk Collaborator • Draft
1
Perfify cost bound pass
#2429 opened Apr 14, 2026 by javathunderman Collaborator • Draft
@javathunderman
1
Fix macos build
#2427 opened Apr 14, 2026 by wsmoses Member Loading…
add diff rule for chlo.square
#2426 opened Apr 14, 2026 by mofeing Collaborator Loading…
1
2
Expand grid options for float types and optimization levels
#2417 opened Apr 9, 2026 by wsmoses Member Loading…
Revert reversions
#2414 opened Apr 9, 2026 by wsmoses Member Loading…
Update jax-ml/jax to commit c12250649c1121e8a20fad33eda8293cf86e75c5 dependencies Pull requests that update a dependency file
#2412 opened Apr 8, 2026 by enzymead-bot Bot Loading…
More multifloat testing
#2409 opened Apr 8, 2026 by wsmoses Member Loading…
1
tpu cp combiner
#2392 opened Apr 5, 2026 by wsmoses Member Loading…
Add support for cudaGetDeviceProperties for later CUDA-12
#2388 opened Apr 3, 2026 by xys-syx Collaborator Loading…
Try sharded atmosphere on TPU
#2370 opened Apr 1, 2026 by Pangoraw Collaborator Loading…
1
8
fix: add a optimization barrier to prevent folding
#2367 opened Mar 31, 2026 by avik-pal Collaborator Loading…
2
Update gb25_commit configuration in workflow
#2363 opened Mar 30, 2026 by avik-pal Collaborator • Draft
Update test-gb-25 workflow with grid options
#2349 opened Mar 29, 2026 by wsmoses Member Loading…
tpu flag debugging
#2348 opened Mar 29, 2026 by wsmoses Member Loading…
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.