Linux-cpp-lisp / opt_einsum_fxLinks
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
☆21Updated 3 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
Sorting:
- Collection of scripts to build PyTorch and the domain libraries from source.☆11Updated this week
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆49Updated 10 months ago
- cuASR: CUDA Algebra for Semirings☆35Updated 2 years ago
- ☆16Updated 2 years ago
- ☆52Updated 9 months ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- CUDA and OpenMP implementations of C2R/R2C inplace transposition☆46Updated 10 years ago
- ☆16Updated 8 months ago
- Automatic High-Order Optimization for Tensors☆23Updated 2 years ago
- Automatically insert nvtx ranges to PyTorch models☆17Updated 4 years ago
- A tool for weighted model counting through tensor network contraction☆21Updated 3 years ago
- An experimental ahead of time compiler for Relay.☆50Updated 5 years ago
- The Combinatorial BLAS (CombBLAS) is an extensible distributed-memory parallel graph library offering a small but powerful set of linear …☆75Updated last week
- A library for syntactically rewriting Python programs, pronounced (sinner).☆69Updated 3 years ago
- Flexible and performant GEMM kernels in Julia☆82Updated 6 months ago
- A Distributed Multi-GPU System for Fast Graph Processing☆65Updated 6 years ago
- A lightweight, Pythonic, frontend for MLIR☆81Updated last year
- ☆104Updated 11 months ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- Compiler with automatic differentiation☆48Updated last year
- JAX for Graphcore IPU (experimental)☆21Updated last year
- Tensor Contraction C++ Library☆53Updated 5 years ago
- Benchmarking OpenBLAS on the Apple M1☆18Updated 4 years ago
- ☆28Updated 4 months ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆84Updated this week
- Codebase associated with the PyTorch compiler tutorial☆45Updated 5 years ago
- Light-weight GPU kernel interface for graph operations☆15Updated 5 years ago
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years ago
- Python bindings for the egg-smol rust library☆76Updated last week