Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
106 Open 1,459 Closed
106 Open 1,459 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

fix: drop more unsupported attributes for tpu lowering
#2088 opened Feb 16, 2026 by avik-pal Loading…
add lit tests for pass ConvertParalleltoGPU2
#2078 opened Feb 13, 2026 by xys-syx Loading…
2
feat: capi for transform dialect passes
#2075 opened Feb 12, 2026 by avik-pal • Draft
2
feat: rewrite if with binary op to a select
#2064 opened Feb 10, 2026 by avik-pal Loading…
1
1
Update jax-ml/jax to commit 1014bf3d7c551921d501da2c857fbef468441cea dependencies Pull requests that update a dependency file
#2052 opened Feb 8, 2026 by enzymead-bot bot Loading…
Bessel lowering
#2051 opened Feb 8, 2026 by jumerckx • Draft
@jumerckx
1
It is a branch that could let the pipeline simple cuda -> amd gpu works, but cannot be merged
#2048 opened Feb 8, 2026 by xys-syx • Draft
Add partial if to select pattern
#2044 opened Feb 7, 2026 by ivanradanov Loading…
4
feat: register cuda tile
#2043 opened Feb 7, 2026 by avik-pal Loading…
Make if_rocm_is_configured to be true and add AMD support for generating binary code
#2037 opened Feb 6, 2026 by xys-syx Loading…
6
Symm lower blas
#2008 opened Jan 29, 2026 by snonk • Draft
3
Multislice spmd
#1981 opened Jan 27, 2026 by wsmoses Loading…
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
add lapack.HeevOp for eigendecomposition
#1941 opened Jan 19, 2026 by mofeing • Draft
1
2
Fix amd target
#1936 opened Jan 15, 2026 by xys-syx • 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 Loading…
17
[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
ProTip! Filter pull requests by the default branch with base:main.