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
  ☆55Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆32Updated 5 years ago
• bdusell / semiring-einsum
  Generic PyTorch implementation of einsum that supports different semirings
  ☆51Updated last month
• microsoft / knossos-ksc
  Compiler with automatic differentiation
  ☆49Updated 2 years ago
• Quansight / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆13Updated 2 months ago
• spcl / npbench
  NPBench - A Benchmarking Suite for High-Performance NumPy
  ☆91Updated last month
• EnzymeAD / Enzyme-JAX
  Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
  ☆109Updated this week
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆43Updated 3 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 5 years ago
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆45Updated last year
• nv-legate / legate.pandas
  An Aspiring Drop-In Replacement for Pandas at Scale
  ☆74Updated 4 years ago
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆205Updated 2 weeks ago
• justindomke / numbat
  NumPy+Jax with named axes and an uncompromising attitude
  ☆23Updated 10 months ago
• hpcgarage / ParTI
  Parallel Tensor Infrastructure (ParTI!)
  ☆33Updated 5 years ago
• microsoft / ADBench
  Benchmarking various AD tools.
  ☆110Updated last year
• manishucsd / py-codegen
  ☆16Updated last year
• SymJAX / SymJAX
  Documentation:
  ☆125Updated 2 years ago
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆164Updated this week
• facebookresearch / loop_tool
  A thin, highly portable toolkit for efficiently compiling dense loop-based computation.
  ☆149Updated 3 years ago
• google-deepmind / einshape
  ☆107Updated last year
• graphcore-research / jax-experimental
  JAX for Graphcore IPU (experimental)
  ☆22Updated last year
• spcl / sten
  Sparsity support for PyTorch
  ☆38Updated 9 months ago
• manya-bansal / mosaic
  ☆14Updated 2 years ago
• Jokeren / triton-samples
  ☆28Updated last year
• JuliaGPU / GemmKernels.jl
  Flexible and performant GEMM kernels in Julia
  ☆83Updated 3 weeks ago
• PASSIONLab / CombBLAS
  The Combinatorial BLAS (CombBLAS) is an extensible distributed-memory parallel graph library offering a small but powerful set of linear …
  ☆80Updated 5 months ago
• JuliaParallel / rodinia
  Julia ports of the Rodinia benchmark suite for heterogeneous computing infrastructures
  ☆55Updated 2 years ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated 2 months ago
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆67Updated 3 years ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 3 years ago