Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
23 Open 346 Closed
23 Open 346 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Raise memref.load to stablehlo.gather
#450 opened Mar 10, 2025 by Pangoraw Loading…
1
Ensure strict affine expr checking
#448 opened Mar 10, 2025 by vimarsh6739 Loading…
4
Split affine.parallels such that we no longer need a nested affine.if
#447 opened Mar 10, 2025 by ivanradanov Loading…
1
Add code for unrolling affine.for under threshold
#441 opened Mar 8, 2025 by snarang181 • Draft
@snarang181
19
vendor affine loop parallelization from upstream in affine-cfg
#429 opened Mar 4, 2025 by ftynse Loading…
2
feat: add side_effect and backend_config to jit_call
#425 opened Mar 3, 2025 by avik-pal Loading…
12
Auto generate StableHLO transform patterns
#422 opened Mar 2, 2025 by GleasonK Loading…
5
WIP Prototype communcation dialect
#368 opened Feb 19, 2025 by EganJ • Draft
Also try numba promotewhile pass
#342 opened Feb 14, 2025 by ivanradanov Loading…
feat: simplify operations with sign
#336 opened Feb 11, 2025 by avik-pal • Draft
7
Lower MPI to stablehlo.custom_call ops
#305 opened Feb 2, 2025 by mofeing • Draft
Add GetDimensionSizeOp as an inactive operation
#269 opened Jan 23, 2025 by sriharikrishna • Draft
WIP Add various llvm raising passes
#265 opened Jan 23, 2025 by ivanradanov Loading…
2
Register context
#232 opened Jan 10, 2025 by wsmoses Loading…
Add "concat of consecutive gathers is replaced with a larger gather"
#207 opened Dec 28, 2024 by glou-nes Loading…
11
fix fft rev diff rule
#176 opened Nov 29, 2024 by mofeing Loading…
1
3
Op like
#174 opened Nov 26, 2024 by wsmoses Loading…
Whiledce
#173 opened Nov 22, 2024 by wsmoses Loading…
Jmp/scatter gather
#142 opened Oct 16, 2024 by wsmoses Loading…
1
More static argnums
#126 opened Sep 5, 2024 by wsmoses Loading…
Now with tpu CI?
#123 opened Sep 3, 2024 by wsmoses Loading…
update llama
#104 opened Jul 18, 2024 by ftynse Loading…
1
Add SelectSimplify op to Enzyme Jax.
#91 opened Jun 17, 2024 by arissa Loading…
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.