EnzymeAD / Enzyme
High-performance automatic differentiation of LLVM and MLIR.
☆1,355Updated this week
Alternatives and similar repositories for Enzyme:
Users that are interested in Enzyme are comparing it to the libraries listed below
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,288Updated 11 months ago
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,300Updated this week
- Extending JAX with custom C++ and CUDA code☆388Updated 7 months ago
- automatic differentiation made easier for C++☆1,740Updated last month
- 21st century AD☆1,510Updated last week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆468Updated this week
- A lightweight high performance tensor algebra framework for modern C++☆776Updated 11 months ago
- Optax is a gradient processing and optimization library for JAX.