Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
45 Open 820 Closed
45 Open 820 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

While dead arg speedup
#1146 opened Jul 5, 2025 by wsmoses Loading…
Start prototyping low-high level QR ops
#1142 opened Jul 2, 2025 by mofeing • Draft
ProbProg: Making trace an operand
#1141 opened Jul 1, 2025 by sbrantq Loading…
ablation tests for jaxmd, neuralgcm
#1137 opened Jul 1, 2025 by mofeing • Draft
[CI] Build wheel on aarch64-linux using lld as linker
#1136 opened Jul 1, 2025 by giordano • Draft
1
1
Restore alternatives lowering
#1116 opened Jun 27, 2025 by wsmoses Loading…
ProbProg: Lowering addSampleToTraceOp to JIT calls
#1002 opened Jun 1, 2025 by sbrantq Loading…
3
feat: general concat reshape to batching pass
#983 opened May 22, 2025 by avik-pal • Draft
3
Ommit block arguments for lift load
#981 opened May 21, 2025 by wsmoses Loading…
1
feat: custom lowering
#963 opened May 18, 2025 by avik-pal • Draft
5
feat: triangular_solve optimizations [part I]
#957 opened May 17, 2025 by avik-pal • Draft
feat: insertdim + dropdim
#903 opened May 4, 2025 by avik-pal • Draft
4
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…
ProTip! no:milestone will show everything without a milestone.