EnzymeAD/Enzyme-JAX

Ver en GitHub

Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.

Estrellas: 131
Forks: 53
Issues abiertos para principiantes: 0
Issues indexados: 291
Merge medio: 1 d 10 h
PR fusionados (30 d): 193

Lenguaje dominante: MLIR
Licencia: Sin datos de licencia
Último push a GitHub: 14/9/2026
Última indexación: 15/9/2026
Guía de contribución: Sin guía de contribución
Código de conducta: Sin código de conducta
Etiquetas para principiantes: good first issue help wanted

Abierto Todos

291 issues abiertos indexados

Add support for `jacrev`, `jacfwd`, `hessian`, `vmap` Abierto

EnzymeAD/Enzyme-JAX#9 · 3 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
HLO Canonicalizations Todo list Abierto

EnzymeAD/Enzyme-JAX#54 · 31 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Tracking issue for missing HLO derivatives Abierto

good first issue help wanted

EnzymeAD/Enzyme-JAX#88 · 4 comentarios · 0 reacciones · 1 asignado · Indexado el 15/9/2026
GPU CI of Jax not thinking its running on GPU Abierto

EnzymeAD/Enzyme-JAX#122 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Profiling API's Abierto

EnzymeAD/Enzyme-JAX#129 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Export issue with missing BLAS symbol Abierto

EnzymeAD/Enzyme-JAX#135 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Local build is broken on macOS Abierto

EnzymeAD/Enzyme-JAX#146 · 3 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Tracking issue for missing Batch Op Interface Abierto

good first issue help wanted

EnzymeAD/Enzyme-JAX#152 · 7 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
TupleOp Abierto

EnzymeAD/Enzyme-JAX#157 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Constprop through chlo.lgamma Abierto

EnzymeAD/Enzyme-JAX#179 · 1 comentario · 0 reacciones · 0 asignados · Indexado el 15/9/2026
An enzyme optimization benchmark Abierto

EnzymeAD/Enzyme-JAX#180 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Optimization Passes for `dynamic_gather` Abierto

EnzymeAD/Enzyme-JAX#184 · 2 comentarios · 1 reacción · 1 asignado · Indexado el 15/9/2026
Missing Transpose Optimizations Abierto

EnzymeAD/Enzyme-JAX#217 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Kernel SROA of arguments Abierto

EnzymeAD/Enzyme-JAX#227 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Incorrect IR generated for Vector Mode AD Abierto

EnzymeAD/Enzyme-JAX#235 · 4 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Kernel Optimizations Abierto

EnzymeAD/Enzyme-JAX#238 · 1 comentario · 0 reacciones · 1 asignado · Indexado el 15/9/2026
`slice(transpose(...))` --> `transpose(slice(...))` if transpose is only being used in slice Abierto

EnzymeAD/Enzyme-JAX#257 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Raise NVVM threadidx to GPU threadidx [and similar] Abierto

EnzymeAD/Enzyme-JAX#274 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
extui and related arithmetic instructions not eliminated Abierto

EnzymeAD/Enzyme-JAX#279 · 3 comentarios · 0 reacciones · 1 asignado · Indexado el 15/9/2026
Scatter Optimizations with Window Dims specified Abierto

EnzymeAD/Enzyme-JAX#308 · 3 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Propagate Constants into Body of Operations Abierto

EnzymeAD/Enzyme-JAX#317 · 3 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
More optimization patterns as required Abierto

EnzymeAD/Enzyme-JAX#370 · 3 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Affine if simplification Abierto

EnzymeAD/Enzyme-JAX#392 · 8 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
GB Stablehlo optimization Abierto

EnzymeAD/Enzyme-JAX#402 · 7 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Add support to raise an affine.for to stablehlo.while Abierto

EnzymeAD/Enzyme-JAX#442 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Pull repeated ops from conditional branches Abierto

EnzymeAD/Enzyme-JAX#444 · 4 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Illegal frem error Abierto

EnzymeAD/Enzyme-JAX#482 · 0 comentarios · 1 reacción · 0 asignados · Indexado el 15/9/2026
CSE of not Abierto

EnzymeAD/Enzyme-JAX#490 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Transpose of a while Abierto

EnzymeAD/Enzyme-JAX#506 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
BatchNorm Inference fusion with Dense or Conv Abierto

EnzymeAD/Enzyme-JAX#533 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
IR right before stablehlo raise Abierto

EnzymeAD/Enzyme-JAX#536 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Remaining memref.load Abierto

EnzymeAD/Enzyme-JAX#543 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Test cases for improved for loop raising Abierto

EnzymeAD/Enzyme-JAX#545 · 7 comentarios · 0 reacciones · 1 asignado · Indexado el 15/9/2026
concat reshape simplification Abierto

EnzymeAD/Enzyme-JAX#557 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Subtract reduction raising Abierto

EnzymeAD/Enzyme-JAX#560 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Further generalize reductions Abierto

EnzymeAD/Enzyme-JAX#566 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Transpose Optimizations Missing (Move Transposes Up) Abierto

EnzymeAD/Enzyme-JAX#568 · 5 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Push reshapes up Abierto

EnzymeAD/Enzyme-JAX#575 · 29 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
ERROR: LoadError: "failed to run pass manager on module" on Reactant script with Oceananigans Abierto

EnzymeAD/Enzyme-JAX#578 · 10 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
DUS creates Allgather operations not removed by XLA Abierto

EnzymeAD/Enzyme-JAX#596 · 10 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Simplify/eliminate DUS Abierto

EnzymeAD/Enzyme-JAX#605 · 3 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Failed raising Abierto

EnzymeAD/Enzyme-JAX#614 · 3 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Padding remains in baroclinic model Abierto

EnzymeAD/Enzyme-JAX#622 · 2 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
`DEADLINE_EXCEEDED: GetKeyValue() timed out` Abierto

EnzymeAD/Enzyme-JAX#623 · 5 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Slice access analysis to eliminate slices and concats inside loops Abierto

EnzymeAD/Enzyme-JAX#625 · 1 comentario · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Unused loop iterarg returning constant Abierto

EnzymeAD/Enzyme-JAX#627 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Slicing to conv Abierto

EnzymeAD/Enzyme-JAX#628 · 9 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Concat of operation with slices on boundary -> dynamic_slice Abierto

EnzymeAD/Enzyme-JAX#630 · 7 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
Conv creates allgather Abierto

EnzymeAD/Enzyme-JAX#653 · 0 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026
OOM for large enough models Abierto

EnzymeAD/Enzyme-JAX#671 · 8 comentarios · 0 reacciones · 0 asignados · Indexado el 15/9/2026

EnzymeAD/Enzyme-JAX

Recibe los nuevos issues en tu correo