Skip to content

Pull requests: EnzymeAD/Enzyme-JAX

New pull request New
64 Open 1,007 Closed
64 Open 1,007 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: TritonCallOp
#1433 opened Sep 27, 2025 by avik-pal • Draft
3
feat: dus to dynamic_pad + dynamic_pad to pad
#1430 opened Sep 26, 2025 by avik-pal Loading…
2
feat: greedy loop to batch
#1425 opened Sep 26, 2025 by avik-pal • Draft
1
Bump hedron hash to fix compile commands issue
#1423 opened Sep 25, 2025 by EganJ Loading…
1
Update JAX to commit 45065d569064392cee45a066f9f788fe29ee2cd8
#1418 opened Sep 23, 2025 by giordano Loading…
8
Cpuify
#1415 opened Sep 23, 2025 by xys-syx Loading…
1
ProbProg: MH op
#1402 opened Sep 20, 2025 by sbrantq • Draft
1
WIP: occupance
#1395 opened Sep 16, 2025 by wsmoses Loading…
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
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
ProTip! Adding no:label will show everything without a label.