EnzymeAD / Enzyme
High-performance automatic differentiation of LLVM and MLIR.
☆1,287Updated this week
Related projects ⓘ
Alternatives and complementary repositories for Enzyme
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,256Updated 7 months ago
- Extending JAX with custom C++ and CUDA code☆378Updated 3 months ago
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,220Updated this week
- clad -- automatic differentiation for C/C++☆291Updated this week
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆863Updated last month
- Numerical linear algebra software package☆410Updated this week
- An Aspiring Drop-In Replacement for NumPy at Scale☆643Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆445Updated 2 weeks ago
- Optax is a gradient processing and optimization library for JAX.☆1,699Updated last week
- A lightweight high performance tensor algebra framework for modern C++☆751Updated 7 months ago
- Important concepts in numerical linear algebra and related areas☆726Updated 10 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆933Updated 2 months ago
- Reinforcement learning environments for compiler and program optimization tasks☆914Updated last month
- common in-memory tensor structure☆912Updated last month
- SuiteSparse graph algorithms in the language of linear algebra. For production: (default) STABLE branch. Code development: ask me for t…☆362Updated this week
- automatic differentiation made easier for C++☆1,658Updated last week
- AutoBound automatically computes upper and lower bounds on functions.☆355Updated 4 months ago
- A code generator for array-based code on CPUs and GPUs☆589Updated this week
- Library for specialized dense and sparse matrix operations, and deep learning primitives.☆850Updated this week
- ☆788Updated this week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆360Updated last month
- Portable and vendor neutral framework for parallel programming on heterogeneous platforms.☆396Updated this week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,123Updated 3 weeks ago
- A pedagogical implementation of Autograd☆955Updated 4 years ago
- SymEngine is a fast symbolic manipulation library, written in C++☆1,169Updated 3 weeks ago
- Distributed multigrid linear solver library on GPU☆495Updated 3 months ago
- A massively-parallel, block-sparse tensor framework written in C++☆259Updated this week
- Clarabel.rs: Interior-point solver for convex conic optimisation problems in Rust.☆340Updated this week
- A header-only C++ library for numerical optimization --☆743Updated this week
- forward and reverse mode automatic differentiation primitives for Julia Base + StdLibs☆436Updated this week