Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
38 Open 625 Closed
38 Open 625 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

feat: sharding aware operation reordering
#859 opened Apr 24, 2025 by avik-pal Loading…
1
chore: bump both enzyme and jax versions
#858 opened Apr 23, 2025 by avik-pal Loading…
13
wip: WhileOp Min cut
#857 opened Apr 23, 2025 by Pangoraw • Draft
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
feat: add side-effect to jit_call
#464 opened Mar 11, 2025 by avik-pal Loading…
10
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
ProTip! What’s not been updated in a month: updated:<2025-03-24.