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++
  ☆31Updated 4 years ago
• microsoft / knossos-ksc
  Compiler with automatic differentiation
  ☆46Updated last year
• iree-org / iree-jax
  ☆51Updated 6 months ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• bdusell / semiring-einsum
  Generic PyTorch implementation of einsum that supports different semirings
  ☆46Updated 7 months ago
• EnzymeAD / Enzyme-JAX
  Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.
  ☆60Updated this week
• google-deepmind / einshape
  ☆100Updated 7 months ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• distdl / distdl
  ☆21Updated 2 years ago
• duvenaud / jaxde
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Updated 5 years ago
• JuliaGPU / GemmKernels.jl
  Flexible and performant GEMM kernels in Julia
  ☆80Updated 3 months ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 3 years ago
• microsoft / tensorflow-rematerialization
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Updated 5 years ago
• srush / learns-dex
  ☆33Updated 4 years ago
• graphcore-research / jax-experimental
  JAX for Graphcore IPU (experimental)
  ☆21Updated 11 months ago
• manishucsd / py-codegen
  ☆15Updated 4 months ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 2 years ago
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• LinjianMa / AutoHOOT
  Automatic High-Order Optimization for Tensors
  ☆23Updated last year
• PhilipVinc / Jax.jl
  ☆28Updated 3 years ago
• JuliaLabs / brutus
  ☆62Updated last week
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆166Updated this week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆37Updated 5 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated last year
• FluxML / XLA.jl
  "Maybe we have our own magic."
  ☆47Updated 4 years ago
• deepchem / jaxchem
  JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling
  ☆79Updated 4 years ago
• metagraph-dev / mlir-graphblas
  MLIR tools and dialect for GraphBLAS
  ☆18Updated 2 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆41Updated 4 years ago
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆55Updated this week