EnzymeAD/Enzyme-JAX

Voir sur GitHub

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

Étoiles: 131
Forks: 53
Issues débutants ouvertes: 0
Issues indexées: 291
Merge moyen: 1 j 10 h
PR mergées (30 j): 193

Langage dominant: MLIR
Licence: Aucune donnée de licence
Dernier push GitHub: 14/9/2026
Dernière indexation: 15/9/2026
Guide de contribution: Aucun guide de contribution
Code de conduite: Aucun code de conduite
Labels débutants: good first issue help wanted

Ouverte Toutes

291 issues ouvertes indexées

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

EnzymeAD/Enzyme-JAX#9 · 3 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
HLO Canonicalizations Todo list Ouverte

EnzymeAD/Enzyme-JAX#54 · 31 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Tracking issue for missing HLO derivatives Ouverte

good first issue help wanted

EnzymeAD/Enzyme-JAX#88 · 4 commentaires · 0 réactions · 1 personne assignée · Indexée le 15/9/2026
GPU CI of Jax not thinking its running on GPU Ouverte

EnzymeAD/Enzyme-JAX#122 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Profiling API's Ouverte

EnzymeAD/Enzyme-JAX#129 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Export issue with missing BLAS symbol Ouverte

EnzymeAD/Enzyme-JAX#135 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Local build is broken on macOS Ouverte

EnzymeAD/Enzyme-JAX#146 · 3 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Tracking issue for missing Batch Op Interface Ouverte

good first issue help wanted

EnzymeAD/Enzyme-JAX#152 · 7 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
TupleOp Ouverte

EnzymeAD/Enzyme-JAX#157 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Constprop through chlo.lgamma Ouverte

EnzymeAD/Enzyme-JAX#179 · 1 commentaire · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
An enzyme optimization benchmark Ouverte

EnzymeAD/Enzyme-JAX#180 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Optimization Passes for `dynamic_gather` Ouverte

EnzymeAD/Enzyme-JAX#184 · 2 commentaires · 1 réaction · 1 personne assignée · Indexée le 15/9/2026
Missing Transpose Optimizations Ouverte

EnzymeAD/Enzyme-JAX#217 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Kernel SROA of arguments Ouverte

EnzymeAD/Enzyme-JAX#227 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Incorrect IR generated for Vector Mode AD Ouverte

EnzymeAD/Enzyme-JAX#235 · 4 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Kernel Optimizations Ouverte

EnzymeAD/Enzyme-JAX#238 · 1 commentaire · 0 réactions · 1 personne assignée · Indexée le 15/9/2026
`slice(transpose(...))` --> `transpose(slice(...))` if transpose is only being used in slice Ouverte

EnzymeAD/Enzyme-JAX#257 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Raise NVVM threadidx to GPU threadidx [and similar] Ouverte

EnzymeAD/Enzyme-JAX#274 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
extui and related arithmetic instructions not eliminated Ouverte

EnzymeAD/Enzyme-JAX#279 · 3 commentaires · 0 réactions · 1 personne assignée · Indexée le 15/9/2026
Scatter Optimizations with Window Dims specified Ouverte

EnzymeAD/Enzyme-JAX#308 · 3 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Propagate Constants into Body of Operations Ouverte

EnzymeAD/Enzyme-JAX#317 · 3 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
More optimization patterns as required Ouverte

EnzymeAD/Enzyme-JAX#370 · 3 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Affine if simplification Ouverte

EnzymeAD/Enzyme-JAX#392 · 8 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
GB Stablehlo optimization Ouverte

EnzymeAD/Enzyme-JAX#402 · 7 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Add support to raise an affine.for to stablehlo.while Ouverte

EnzymeAD/Enzyme-JAX#442 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Pull repeated ops from conditional branches Ouverte

EnzymeAD/Enzyme-JAX#444 · 4 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Illegal frem error Ouverte

EnzymeAD/Enzyme-JAX#482 · 0 commentaires · 1 réaction · 0 personnes assignées · Indexée le 15/9/2026
CSE of not Ouverte

EnzymeAD/Enzyme-JAX#490 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Transpose of a while Ouverte

EnzymeAD/Enzyme-JAX#506 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
BatchNorm Inference fusion with Dense or Conv Ouverte

EnzymeAD/Enzyme-JAX#533 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
IR right before stablehlo raise Ouverte

EnzymeAD/Enzyme-JAX#536 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Remaining memref.load Ouverte

EnzymeAD/Enzyme-JAX#543 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Test cases for improved for loop raising Ouverte

EnzymeAD/Enzyme-JAX#545 · 7 commentaires · 0 réactions · 1 personne assignée · Indexée le 15/9/2026
concat reshape simplification Ouverte

EnzymeAD/Enzyme-JAX#557 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Subtract reduction raising Ouverte

EnzymeAD/Enzyme-JAX#560 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Further generalize reductions Ouverte

EnzymeAD/Enzyme-JAX#566 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Transpose Optimizations Missing (Move Transposes Up) Ouverte

EnzymeAD/Enzyme-JAX#568 · 5 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Push reshapes up Ouverte

EnzymeAD/Enzyme-JAX#575 · 29 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
ERROR: LoadError: "failed to run pass manager on module" on Reactant script with Oceananigans Ouverte

EnzymeAD/Enzyme-JAX#578 · 10 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
DUS creates Allgather operations not removed by XLA Ouverte

EnzymeAD/Enzyme-JAX#596 · 10 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Simplify/eliminate DUS Ouverte

EnzymeAD/Enzyme-JAX#605 · 3 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Failed raising Ouverte

EnzymeAD/Enzyme-JAX#614 · 3 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Padding remains in baroclinic model Ouverte

EnzymeAD/Enzyme-JAX#622 · 2 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
`DEADLINE_EXCEEDED: GetKeyValue() timed out` Ouverte

EnzymeAD/Enzyme-JAX#623 · 5 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Slice access analysis to eliminate slices and concats inside loops Ouverte

EnzymeAD/Enzyme-JAX#625 · 1 commentaire · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Unused loop iterarg returning constant Ouverte

EnzymeAD/Enzyme-JAX#627 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Slicing to conv Ouverte

EnzymeAD/Enzyme-JAX#628 · 9 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Concat of operation with slices on boundary -> dynamic_slice Ouverte

EnzymeAD/Enzyme-JAX#630 · 7 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
Conv creates allgather Ouverte

EnzymeAD/Enzyme-JAX#653 · 0 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026
OOM for large enough models Ouverte

EnzymeAD/Enzyme-JAX#671 · 8 commentaires · 0 réactions · 0 personnes assignées · Indexée le 15/9/2026

EnzymeAD/Enzyme-JAX

Recevez les nouvelles issues par e-mail