Linux-cpp-lisp / opt_einsum_fx
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
☆19Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for opt_einsum_fx
- Generic PyTorch implementation of einsum that supports different semirings☆46Updated 4 months ago
- Flexible and performant GEMM kernels in Julia☆80Updated this week
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- ☆49Updated this week
- Compiler with automatic differentiation☆45Updated last year
- ☆48Updated 3 months ago
- ☆60Updated 10 months ago
- ☆99Updated 5 months ago
- ☆29Updated 2 years ago
- ☆14Updated 2 months ago
- Generic API for dispatch to Pyro backends.☆16Updated 2 years ago
- A tracing JIT compiler for PyTorch☆12Updated 2 years ago
- MLIR tools and dialect for GraphBLAS☆17Updated 2 years ago
- Yet Another Source to Source Automatic Differentiation☆26Updated 5 years ago
- ☆20Updated 3 weeks ago
- Automatic High-Order Optimization for Tensors☆23Updated last year
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆73Updated this week
- POC work on MLIR backend☆50Updated 3 months ago
- Automatic Differentiation for Tensor Algebras☆28Updated 6 years ago
- A LinearOperator implementation for PyTorch☆18Updated 3 years ago
- Probabilistic Circuits from the Juice library☆104Updated 5 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago
- ☆22Updated last year
- "Maybe we have our own magic."☆47Updated 4 years ago
- cuASR: CUDA Algebra for Semirings☆34Updated 2 years ago
- Tools for JAX☆43Updated last week
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆31Updated 2 years ago
- Hierarchical variational models for physics.☆18Updated 4 years ago
- Scalable Probabilistic Programming Library☆21Updated 6 years ago