EnzymeAD / EnzymeLinks
High-performance automatic differentiation of LLVM and MLIR.
☆1,439Updated this week
Alternatives and similar repositories for Enzyme
Users that are interested in Enzyme are comparing it to the libraries listed below
Sorting:
- clad -- automatic differentiation for C/C++☆350Updated this week
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,315Updated 3 months ago
- 21st century AD☆1,536Updated last month
- Numerical linear algebra software package☆486Updated this week
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,343Updated this week
- Extending JAX with custom C++ and CUDA code☆398Updated 11 months ago
- automatic differentiation made easier for C++☆1,824Updated 6 months ago
- common in-memory tensor structure☆1,044Updated last month
- CUDA programming in Julia.☆1,305Updated last week
- A lightweight high performance tensor algebra framework for modern C++☆797Updated last month
- Backward compatible ML compute opset inspired by HLO/MHLO☆514Updated 2 weeks ago
- SuiteSparse graph algorithms in the language of linear algebra. For production: (default) STABLE branch. Code development: ask me for t…☆384Updated this week
- The Legion Parallel Programming System☆737Updated last month
- A code generator for array-based code on CPUs and GPUs☆609Updated last week
- NumPy and SciPy on Multi-Node Multi-GPU systems☆922Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆489Updated last week
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆935Updated last month
- A Python compiler design toolkit.☆384Updated this week
- AutoBound automatically computes upper and lower bounds on functions.☆361Updated last year
- Optax is a gradient processing and optimization library for JAX.☆1,970Updated last week
- Important concepts in numerical linear algebra and related areas☆774Updated last year
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,603Updated this week
- DaCe - Data Centric Parallel Programming☆546Updated this week
- Reinforcement learning environments for compiler and program optimization tasks☆962Updated 10 months ago
- Distributed multigrid linear solver library on GPU☆581Updated 6 months ago
- Library for specialized dense and sparse matrix operations, and deep learning primitives.☆889Updated this week
- Portable and vendor neutral framework for parallel programming on heterogeneous platforms.☆426Updated last month
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆985Updated 3 months ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆461Updated 2 weeks ago
- MLIR For Beginners tutorial☆1,037Updated 3 weeks ago