Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
60 Open 982 Closed
60 Open 982 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

WIP: occupance
#1395 opened Sep 16, 2025 by wsmoses Loading…
Bump actions/download-artifact from 4 to 5 dependencies Pull requests that update a dependency file github_actions Pull requests that update GitHub Actions code
#1392 opened Sep 15, 2025 by dependabot bot Loading…
1
feat: batching code based slice ops
#1386 opened Sep 11, 2025 by avik-pal Loading…
12 tasks done
11
Use larger grid on full GB-25 test
#1365 opened Sep 4, 2025 by giordano • Draft
Try using select instead of add/mul
#1363 opened Sep 3, 2025 by wsmoses Loading…
6
Test Disable comm to pad transforms
#1360 opened Sep 3, 2025 by wsmoses Loading…
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 • Draft
31
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
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
ablation tests for jaxmd, neuralgcm
#1137 opened Jul 1, 2025 by mofeing • Draft
Restore alternatives lowering
#1116 opened Jun 27, 2025 by wsmoses Loading…
ProbProg: Lowering addSampleToTraceOp to JIT calls
#1002 opened Jun 1, 2025 by sbrantq Loading…
3
ProTip! Follow long discussions with comments:>50.