EnzymeAD / Enzyme

Related projects ⓘ

Alternatives and complementary repositories for Enzyme

• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,254Updated 6 months ago
• dfm / extending-jax
  Extending JAX with custom C++ and CUDA code
  ☆379Updated 2 months ago
• vgvassilev / clad
  clad -- automatic differentiation for C/C++
  ☆288Updated this week
• nv-legate / cunumeric
  An Aspiring Drop-In Replacement for NumPy at Scale
  ☆621Updated last month
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,211Updated this week
• autodiff / autodiff
  automatic differentiation made easier for C++
  ☆1,653Updated 2 weeks ago
• dgasmith / opt_einsum
  ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
  ☆856Updated last month
• inducer / loopy
  A code generator for array-based code on CPUs and GPUs
  ☆587Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆905Updated last month
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆439Updated this week
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆930Updated last month
• ginkgo-project / ginkgo
  Numerical linear algebra software package
  ☆406Updated this week
• NVIDIA / cuCollections
  ☆485Updated this week
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆408Updated this week
• nv-legate / legate.core
  The Foundation for All Legate Libraries
  ☆189Updated last month
• libxsmm / libxsmm
  Library for specialized dense and sparse matrix operations, and deep learning primitives.
  ☆850Updated this week
• romeric / Fastor
  A lightweight high performance tensor algebra framework for modern C++
  ☆751Updated 6 months ago
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆2,690Updated this week
• google-deepmind / chex
  ☆785Updated last week
• mattjj / autodidact
  A pedagogical implementation of Autograd
  ☆954Updated 4 years ago
• google / autobound
  AutoBound automatically computes upper and lower bounds on functions.
  ☆355Updated 4 months ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,350Updated this week
• patrick-kidger / lineax
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆353Updated 2 weeks ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,687Updated this week
• symengine / symengine
  SymEngine is a fast symbolic manipulation library, written in C++
  ☆1,165Updated last week
• libocca / occa
  Portable and vendor neutral framework for parallel programming on heterogeneous platforms.
  ☆393Updated 3 months ago
• higham / what-is
  Important concepts in numerical linear algebra and related areas
  ☆726Updated 9 months ago
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,103Updated last week
• DrTimothyAldenDavis / GraphBLAS
  SuiteSparse graph algorithms in the language of linear algebra. For production: (default) STABLE branch. Code development: ask me for t…
  ☆360Updated this week
• google / ml-compiler-opt
  Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.
  ☆632Updated last week