Alternatives and similar repositories for opt_einsum_fx

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

• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆32Updated 5 years ago
• iree-org / iree-jax
  ☆55Updated last year
• microsoft / knossos-ksc
  Compiler with automatic differentiation
  ☆49Updated 2 years ago
• spcl / npbench
  NPBench - A Benchmarking Suite for High-Performance NumPy
  ☆91Updated last week
• bdusell / semiring-einsum
  Generic PyTorch implementation of einsum that supports different semirings
  ☆51Updated 2 months ago
• manya-bansal / mosaic
  ☆14Updated 2 years ago
• openteams-ai / torch-build
  Collection of scripts to build PyTorch and the domain libraries from source.
  ☆13Updated this week
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆44Updated 3 years ago
• egraphs-good / egglog-python
  Python bindings for the egglog rust library
  ☆96Updated this week
• manishucsd / py-codegen
  ☆16Updated last year
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆45Updated last year
• SymJAX / SymJAX
  Documentation:
  ☆126Updated 2 years ago
• vardigroup / TensorOrder
  A tool for weighted model counting through tensor network contraction
  ☆21Updated 4 years ago
• graphcore-research / jax-experimental
  JAX for Graphcore IPU (experimental)
  ☆22Updated last year
• EnzymeAD / Enzyme-JAX
  Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
  ☆112Updated this week
• duvenaud / jaxde
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Updated 6 years ago
• facebookresearch / loop_tool
  A thin, highly portable toolkit for efficiently compiling dense loop-based computation.
  ☆149Updated 3 years ago
• saforem2 / l2hmc-qcd
  Application of the L2HMC algorithm to simulations in lattice QCD.
  ☆68Updated 2 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 4 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 5 years ago
• spcl / sten
  Sparsity support for PyTorch
  ☆38Updated 10 months ago
• PhilipVinc / Jax.jl
  ☆28Updated 4 years ago
• JuliaGPU / GemmKernels.jl
  Flexible and performant GEMM kernels in Julia
  ☆83Updated this week
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆85Updated last month
• HicrestLaboratory / SPARTA
  SParse AcceleRation on Tensor Architecture
  ☆18Updated 10 months ago
• metagraph-dev / mlir-graphblas
  MLIR tools and dialect for GraphBLAS
  ☆18Updated 3 years ago
• pyprob / ppx
  Probabilistic Programming eXecution protocol (PPX)
  ☆76Updated 3 years ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated 3 years ago
• pyro-ppl / pyro-api
  Generic API for dispatch to Pyro backends.
  ☆16Updated 3 years ago
• bryancatanzaro / inplace
  CUDA and OpenMP implementations of C2R/R2C inplace transposition
  ☆48Updated 10 years ago