EnzymeAD / Enzyme
High-performance automatic differentiation of LLVM and MLIR.
☆1,337Updated 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,277Updated 10 months ago
- 21st century AD☆1,499Updated 3 weeks ago
- clad -- automatic differentiation for C/C++☆308Updated this week
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,261Updated last week
- common in-memory tensor structure☆942Updated 2 weeks ago
- Extending JAX with custom C++ and CUDA code☆386Updated 6 months ago
- Library for specialized dense and sparse matrix operations, and deep learning primitives.☆862Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆462Updated 3 weeks ago
- Backward compatible ML compute opset inspired by HLO/MHLO☆446Updated last week
- CUDA programming in Julia.☆1,250Updated this week
- An Aspiring Drop-In Replacement for NumPy at Scale☆827Updated this week
- A code generator for array-based code on CPUs and GPUs☆598Updated this week
- Research language for array processing in the Haskell/ML family☆1,603Updated 3 weeks ago
- Numerical linear algebra software package☆432Updated this week
- automatic differentiation made easier for C++☆1,725Updated 3 weeks ago
- ☆515Updated this week
- A lightweight high performance tensor algebra framework for modern C++☆771Updated 10 months ago
- A Python Compiler Design Toolkit☆312Updated this week
- Distributed multigrid linear solver library on GPU☆527Updated 2 weeks ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆885Updated 4 months ago
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,424Updated this week
- CUDA Core Compute Libraries☆1,468Updated this week
- AutoBound automatically computes upper and lower bounds on functions.☆359Updated 7 months ago
- C/C++ frontend for MLIR. Also features polyhedral optimizations, parallel optimizations, and more!☆511Updated 4 months ago
- Optax is a gradient processing and optimization library for JAX.☆1,802Updated this week
- A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).☆521Updated this week
- The Legion Parallel Programming System☆707Updated last month
- egg is a flexible, high-performance e-graph library☆1,431Updated last month
- The Foundation for All Legate Libraries☆204Updated last week
- Reinforcement learning environments for compiler and program optimization tasks☆932Updated 4 months ago