Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
95 Open 1,327 Closed
95 Open 1,327 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

Generalize split parallel inductions
#1904 opened Jan 7, 2026 by wsmoses Loading…
feat: more optimizations
#1903 opened Jan 7, 2026 by avik-pal • Draft
2 of 3 tasks
feat: fold constants above certain threshold
#1902 opened Jan 7, 2026 by avik-pal Loading…
4
[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
feat: transpose scatter to scatter transpose
#1887 opened Jan 5, 2026 by avik-pal • Draft
1
add amd gpu module
#1875 opened Jan 4, 2026 by xys-syx • Draft
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
MPI Ops and lowering passes
#1829 opened Dec 23, 2025 by romanlee Loading…
14
Update README with Usage Examples section
#1801 opened Dec 18, 2025 by google-labs-jules bot • Draft
2
Update jax-ml/jax to commit 816e644cc4eb1a4353894ba52d5e0d92716936e5 dependencies Pull requests that update a dependency file
#1791 opened Dec 17, 2025 by enzymead-bot bot Loading…
1
Add pass to convert atomic rmw to non-atomic ops when legal
#1782 opened Dec 16, 2025 by ivanradanov Loading…
2
Optimize consecutive DUS operations with slices from same source to single DUS with extend
#1768 opened Dec 15, 2025 by Copilot AI • Draft
1
@wsmoses
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
Add adaptToCallingConvention utility for element type conversion and byte offset handling
#1709 opened Dec 8, 2025 by Copilot AI • Draft
1
@wsmoses
13
feat: add outside values to while loop argument list
#1673 opened Dec 2, 2025 by avik-pal Loading…
5
Add partial symmetry detection
#1663 opened Nov 28, 2025 by gaurav-arya Loading…
2
feat: triton lowering [take III]
#1660 opened Nov 26, 2025 by avik-pal • Draft
1
Symm lowering
#1659 opened Nov 26, 2025 by snonk • Draft
5
Add copilot-instructions.md for repository onboarding
#1657 opened Nov 25, 2025 by Copilot AI Loading…
@avik-pal
2
Add ReverseConstProp pattern to fold reverse of constant tensors
#1655 opened Nov 25, 2025 by Copilot AI • Draft
10 tasks done
1
@wsmoses
18
ProTip! Mix and match filters to narrow down what you’re looking for.