Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
66 Open 921 Closed
66 Open 921 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

ci: upload test timings & add comment
#1328 opened Aug 20, 2025 by avik-pal Loading…
Format code of branch "main" format
#1327 opened Aug 20, 2025 by github-actions bot Loading…
Increase tolerance for jaxmd test
#1326 opened Aug 19, 2025 by giordano Loading…
1
Post-commit addressing review comments
#1324 opened Aug 19, 2025 by tyb0807 Loading…
Add broken canonicalize scf for case
#1323 opened Aug 19, 2025 by wsmoses Loading…
for: Add rewrite pattern to apply loop attributes
#1321 opened Aug 19, 2025 by Pangoraw Loading…
6
ExtendSlice 2
#1319 opened Aug 19, 2025 by jumerckx Loading…
1
1
Bump actions/checkout from 4 to 5 dependencies Pull requests that update a dependency file github_actions Pull requests that update GitHub Actions code
#1315 opened Aug 18, 2025 by dependabot bot • Draft
1
Update the xla commit to experiment potential fix for OOM
#1307 opened Aug 15, 2025 by felixwqp Loading…
19
feat: Support for Add operation in AutoDiffScatterRev
#1299 opened Aug 12, 2025 by JulianTrommer Loading…
1
ops for SVD factorization
#1291 opened Aug 9, 2025 by mofeing • Draft
small fixes for QR ops lowering to custom_call
#1290 opened Aug 9, 2025 by mofeing Loading…
1
feat: detect and rewrite elementwise as reduce_window/conv
#1274 opened Aug 7, 2025 by avik-pal Loading…
6
SplitMultiResultScatter
#1260 opened Aug 6, 2025 by jumerckx Loading…
1
1
Emit alternatives to pass option [WIP]
#1256 opened Aug 5, 2025 by wsmoses Loading…
1
let's check out tpu ci
#1242 opened Aug 2, 2025 by wsmoses Loading…
Try enabling jaxmd rev
#1190 opened Jul 28, 2025 by wsmoses Loading…
Try enabling segment anything
#1189 opened Jul 28, 2025 by wsmoses Loading…
Try enabling bert
#1188 opened Jul 28, 2025 by wsmoses Loading…
Try enabling mistral
#1187 opened Jul 28, 2025 by wsmoses Loading…
Try enabling stable diffusion
#1186 opened Jul 28, 2025 by wsmoses Loading…
Try enabling maxtext benchmarking
#1185 opened Jul 28, 2025 by wsmoses Loading…
feat: stablehlo in triton kernels
#1171 opened Jul 23, 2025 by avik-pal • Draft
feat: derivative rules for enzymexla.ml ops
#1150 opened Jul 9, 2025 by avik-pal • Draft
ProbProg: Making trace an operand
#1141 opened Jul 1, 2025 by sbrantq Loading…
6
ProTip! Exclude everything labeled bug with -label:bug.