Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
270 Open 2,157 Closed
270 Open 2,157 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

CuDNNHLOOpt: don't fuse a dot_general with no contracting dimensions
#3059 opened Sep 1, 2026 by AsafManela Contributor Loading…
1
Eganj distributed higher level kernels
#3055 opened Sep 1, 2026 by EganJ Collaborator • Draft
1
refactor MPI constant binding with a generic API
#3052 opened Sep 1, 2026 by mofeing Collaborator Loading…
reuse JIT addresses for MPI FFI handler function bindings
#3051 opened Sep 1, 2026 by mofeing Collaborator Loading…
1
Raise some affine ifs to DUS
#3050 opened Sep 1, 2026 by Pangoraw Collaborator Loading…
1
Only apply auto-batching to chains that can be fully hoisted
#3049 opened Sep 1, 2026 by Pangoraw Collaborator • Draft
Update jax-ml/jax to commit 3b3bebe4d863f5a7cfbe99c6709b6c28aca1bb65 dependencies Pull requests that update a dependency file
#3042 opened Aug 31, 2026 by enzymead-bot Bot Loading…
1
llvm-to-affine-access: split stores through selected views into the arms
#3032 opened Aug 31, 2026 by wsmoses Member Loading…
Raising: expand buffer-yielding branches into their accesses
#3027 opened Aug 30, 2026 by wsmoses Member Loading…
Raising: forward the packed capture struct; distribute geps over selects
#3026 opened Aug 30, 2026 by wsmoses Member Loading…
Raising: carry atomic adds through viewed geps and raw accesses
#3025 opened Aug 30, 2026 by wsmoses Member Loading…
EnzymeHLOOpt: scatter to DUS only when the index is provably in bounds
#3022 opened Aug 29, 2026 by wsmoses Member Loading…
feat: support raising recurrence relations
#3020 opened Aug 29, 2026 by avik-pal Collaborator • Draft
1
Raising: expand affine.if pointer yields into their accesses
#3016 opened Aug 29, 2026 by wsmoses Member Loading…
1
Raising: chase register-array scratch
#3015 opened Aug 29, 2026 by wsmoses Member Loading…
Raising: unroll invariant-yield do-whiles
#3013 opened Aug 28, 2026 by wsmoses Member Loading…
Raising: split struct scratch and flatten viewed scratch
#3012 opened Aug 28, 2026 by wsmoses Member Loading…
Raising: rewrite pointer induction to indexed accesses
#3011 opened Aug 28, 2026 by wsmoses Member Loading…
Raising: fold reads of constant globals; reject host-alloca kernel operands
#3005 opened Aug 28, 2026 by wsmoses Member Loading…
Raising: unroll budget for constant-bound loops
#3004 opened Aug 28, 2026 by wsmoses Member Loading…
While raising: broadcast under-attributed yields to the carried layout
#3002 opened Aug 28, 2026 by wsmoses Member Loading…
1
Raising: canonicalize maps before permutation matching; while fallback for constant-bound loops
#2998 opened Aug 28, 2026 by wsmoses Member Loading…
Raising: privatize lane-local scratch
#2996 opened Aug 27, 2026 by wsmoses Member Loading…
1
CommonCompareExpressionRewrite: place the negation after the op it reads
#2995 opened Aug 27, 2026 by wsmoses Member Loading…
Raising: dead lanes must not scatter at all
#2993 opened Aug 27, 2026 by wsmoses Member Loading…
4
ProTip! Adding no:label will show everything without a label.