Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
93 Open 1,349 Closed
93 Open 1,349 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 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
add a pattern to remove idempotent DUS from a while
#1921 opened Jan 9, 2026 by ftynse Loading…
Implementation of periodic checkpointing
#1910 opened Jan 8, 2026 by sriharikrishna • Draft
3
[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
add amd gpu module
#1875 opened Jan 4, 2026 by xys-syx • 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
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
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
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
ProTip! Exclude everything labeled bug with -label:bug.