Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
102 Open 1,397 Closed
102 Open 1,397 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

add value for amd chips and abi version
#2005 opened Jan 28, 2026 by xys-syx Loading…
5
raising: better detect need for scatter/gather
#1997 opened Jan 28, 2026 by Pangoraw Loading…
1
Multislice spmd
#1981 opened Jan 27, 2026 by wsmoses Loading…
Add LowerMultiRotateOp and LowerMultiSliceOp
#1974 opened Jan 26, 2026 by jumerckx Loading…
@jumerckx
8
Optimize FFT of zero tensors to zero constant
#1952 opened Jan 21, 2026 by Copilot AI • Draft
1
@wsmoses
Implement MultiRotate and MultiSlice ops with optimization patterns
#1950 opened Jan 21, 2026 by Copilot AI • Draft
1
@wsmoses
feat: expand batchnorm ops if needed
#1946 opened Jan 20, 2026 by avik-pal • Draft
ProbProg: MCMC
#1943 opened Jan 20, 2026 by sbrantq • Draft
add lapack.HeevOp for eigendecomposition
#1941 opened Jan 19, 2026 by mofeing • Draft
1
Fix amd target
#1936 opened Jan 15, 2026 by xys-syx • Draft
WIP DusDusPadPad
#1931 opened Jan 12, 2026 by ftynse • Draft
build(deps): bump bazel-contrib/setup-bazel from 0.16.0 to 0.18.0 dependencies Pull requests that update a dependency file github_actions Pull requests that update GitHub Actions code
#1930 opened Jan 12, 2026 by dependabot bot Loading…
1
Add DUSDUSPadPadToDUSPad optimization to merge consecutive padded dynamic update slices
#1929 opened Jan 10, 2026 by Copilot AI • Draft
@wsmoses
Optimize DUS of select with slice from same source or operand
#1927 opened Jan 9, 2026 by Copilot AI • Draft
20 of 21 tasks
1
@wsmoses
8
Implementation of periodic checkpointing
#1910 opened Jan 8, 2026 by sriharikrishna • Draft
12
[CI] Further improve gb25 caching
#1901 opened Jan 7, 2026 by wsmoses Loading…
3
test: add jaxley benchmark
#1896 opened Jan 7, 2026 by avik-pal • Draft
2
feat: transpose scatter to scatter transpose
#1887 opened Jan 5, 2026 by avik-pal • Draft
1
feat: raise stencils to reduce_window
#1874 opened Dec 31, 2025 by avik-pal Loading…
2
feat: raise scan like operation to reduce window
#1872 opened Dec 30, 2025 by avik-pal • Draft
fix: neural gcm enable correctness check
#1848 opened Dec 25, 2025 by avik-pal • Draft
Update README with Usage Examples section
#1801 opened Dec 18, 2025 by google-labs-jules bot • Draft
2
Fold negate operations on constants by moving UnaryConstProp to high priority
#1733 opened Dec 11, 2025 by Copilot AI • Draft
1
@wsmoses
8
Extend DotGeneral CSE to recognize operand-swapped equivalence
#1726 opened Dec 11, 2025 by Copilot AI • Draft
8 tasks done
1
@avik-pal
2
Optimize dynamic_gather operations on transposed tensors
#1712 opened Dec 8, 2025 by Copilot AI • Draft
4 of 5 tasks
1
@avik-pal
2
ProTip! Adding no:label will show everything without a label.