Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
91 Open 1,313 Closed
91 Open 1,313 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

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
feat: remove more intermediate reshape operations
#1870 opened Dec 29, 2025 by avik-pal Loading…
2
4
Extend Rotate recognition
#1856 opened Dec 27, 2025 by wsmoses Loading…
4
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…
10
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
Update Without Corners
#1757 opened Dec 14, 2025 by wsmoses Loading…
1
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
1
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
Add reverse(reverse(x)) -> x optimization pattern
#1654 opened Nov 25, 2025 by Copilot AI • Draft
1
@wsmoses
feat: general structured matrix lattice
#1629 opened Nov 18, 2025 by avik-pal • Draft
2
add recursive symmetry checks + check for existing attr
#1626 opened Nov 17, 2025 by gaurav-arya Loading…
11
feat: lowering Triton Call triton
#1605 opened Nov 11, 2025 by avik-pal • Draft
2
ProTip! Add no:assignee to see everything that’s not assigned.