Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
32 Open 516 Closed
32 Open 516 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

Support recognizing wrap op with reshapes
#711 opened Apr 9, 2025 by ivanradanov • Draft
Extend op recognition improvements
#710 opened Apr 9, 2025 by ivanradanov Loading…
Possible MacOS CI fix?
#694 opened Apr 8, 2025 by vimarsh6739 Loading…
5
pass to split huge blocks, may be helpful with compile time
#686 opened Apr 7, 2025 by ftynse Loading…
CompileTime: fix reshape elementwise
#680 opened Apr 7, 2025 by wsmoses Loading…
WIP: sumtodot
#656 opened Apr 6, 2025 by wsmoses • Draft
WIP
#590 opened Apr 1, 2025 by chelini Loading…
Gather To Slice op in stableHlo
#550 opened Mar 29, 2025 by arpitj1 • Draft
If Index cast optimization
#544 opened Mar 28, 2025 by arpitj1 Loading…
4
WIP: select-to-if
#540 opened Mar 27, 2025 by wsmoses Loading…
Split parallel iteration space to simplify if statements
#497 opened Mar 19, 2025 by ivanradanov • Draft
1
feat: add side-effect to jit_call
#464 opened Mar 11, 2025 by avik-pal Loading…
4
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
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
ProTip! Add no:assignee to see everything that’s not assigned.