Linux-cpp-lisp / opt_einsum_fx
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
☆19Updated 2 years ago
Alternatives and similar repositories for opt_einsum_fx:
Users that are interested in opt_einsum_fx are comparing it to the libraries listed below
- Generic PyTorch implementation of einsum that supports different semirings☆46Updated 6 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Compiler with automatic differentiation☆45Updated last year
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ☆50Updated 5 months ago
- Flexible and performant GEMM kernels in Julia☆80Updated last month
- Automatic High-Order Optimization for Tensors☆23Updated last year
- ☆53Updated this week
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- ☆28Updated 3 years ago
- ☆22Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- cuASR: CUDA Algebra for Semirings☆35Updated 2 years ago
- ☆54Updated last year
- A tracing JIT compiler for PyTorch☆12Updated 3 years ago
- Generic API for dispatch to Pyro backends.☆16Updated 2 years ago
- ☆100Updated 6 months ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆76Updated 2 months ago
- Stencil computations in JAX☆70Updated last year
- Jax Decompiler☆13Updated 5 months ago
- Sparsity support for PyTorch☆33Updated last month
- ☆16Updated last month
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- Unitary Evolution Recurrent Neural Network☆25Updated 3 years ago
- Tools for JAX☆44Updated last week
- MLIR tools and dialect for GraphBLAS☆18Updated 2 years ago
- JAX for Graphcore IPU (experimental)☆21Updated 10 months ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago