Alternatives and similar repositories for opt_einsum_fx

Users that are interested in opt_einsum_fx are comparing it to the libraries listed below

• iree-org / iree-jax
  ☆51Updated 9 months ago
• Quansight / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆10Updated last week
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 5 years ago
• google-deepmind / einshape
  ☆103Updated 10 months ago
• bdusell / semiring-einsum
  Generic PyTorch implementation of einsum that supports different semirings
  ☆48Updated 10 months ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆35Updated 2 years ago
• numba / numba-mlir
  POC work on MLIR backend
  ☆55Updated 8 months ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 3 years ago
• microsoft / knossos-ksc
  Compiler with automatic differentiation
  ☆48Updated last year
• bryancatanzaro / inplace
  CUDA and OpenMP implementations of C2R/R2C inplace transposition
  ☆46Updated 10 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 2 years ago
• JuliaGPU / GemmKernels.jl
  Flexible and performant GEMM kernels in Julia
  ☆82Updated 5 months ago
• Jokeren / triton-samples
  ☆27Updated 4 months ago
• EnzymeAD / Enzyme-JAX
  Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
  ☆78Updated this week
• LinjianMa / AutoHOOT
  Automatic High-Order Optimization for Tensors
  ☆23Updated 2 years ago
• manishucsd / py-codegen
  ☆16Updated 7 months ago
• PhilipVinc / Jax.jl
  ☆28Updated 3 years ago
• pengmai / lagrad
  An MLIR-based source-to-source automatic differentiation system.
  ☆12Updated 2 years ago
• spcl / npbench
  NPBench - A Benchmarking Suite for High-Performance NumPy
  ☆81Updated this week
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆40Updated 8 months ago
• springer13 / tcl
  Tensor Contraction C++ Library
  ☆52Updated 5 years ago
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆28Updated last year
• graphcore-research / jax-experimental
  JAX for Graphcore IPU (experimental)
  ☆21Updated last year
• microsoft / tensorflow-rematerialization
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Updated 5 years ago
• edin-dal / structtensor
  ☆19Updated 6 months ago
• HPAC / tccg
  Tensor Contraction Code Generator
  ☆37Updated 7 years ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• buwantaiji / FermiFlow
  Ab-initio simulation of interacting fermions with equivariant normalizing flow.
  ☆34Updated 2 years ago