Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
96 Open 1,338 Closed
96 Open 1,338 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: compare op simplification for non-negative operands
#1916 opened Jan 8, 2026 by avik-pal Loading…
fix: use AP* for iota check instead of double
#1914 opened Jan 8, 2026 by avik-pal Loading…
DUSDUS to extend
#1911 opened Jan 8, 2026 by wsmoses Loading…
Implementation of periodic checkpointing
#1910 opened Jan 8, 2026 by sriharikrishna • Draft
1
feat: more optimizations
#1903 opened Jan 7, 2026 by avik-pal • Draft
2 of 4 tasks
[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
1
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…
15
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
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
ProTip! no:milestone will show everything without a milestone.