Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
35 Open 653 Closed
35 Open 653 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

Nanobind
#896 opened May 3, 2025 by wsmoses Loading…
feat: factorization ops
#887 opened May 1, 2025 by avik-pal • Draft
3
Cleanup concat wrap
#835 opened Apr 15, 2025 by wsmoses Loading…
Fixing comms by subtracting the inside
#827 opened Apr 15, 2025 by wsmoses Loading…
[Experiment] No sharding for constants to pad
#826 opened Apr 15, 2025 by wsmoses Loading…
Use shardy const for sharded constants
#825 opened Apr 15, 2025 by wsmoses Loading…
WIP select based choice
#796 opened Apr 13, 2025 by wsmoses Loading…
Support right to left rotate
#739 opened Apr 11, 2025 by ivanradanov • Draft
7
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
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
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
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.