Einsum optimization using opt_einsum and PyTorch FX graph rewriting
☆22Mar 17, 2022Updated 4 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:
- SMT-LIB benchmarks for shape computations from deep learning models in PyTorch☆18Dec 21, 2022Updated 3 years ago
- ☆14May 21, 2024Updated last year
- Julia package for quantum circuit generation☆14Jan 22, 2022Updated 4 years ago
- Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code☆10Aug 29, 2023Updated 2 years ago
- Light-weight GPU kernel interface for graph operations☆15May 20, 2020Updated 5 years ago
- ucas hpc course code☆15May 24, 2023Updated 2 years ago
- ☆10Sep 27, 2024Updated last year
- Super Monkey Ball 2 Practice Mod☆19Sep 27, 2025Updated 5 months ago
- Molecular integrals over Gaussian basis functions using sympy.☆16Oct 2, 2024Updated last year
- NVIDIA GPU direct RDMA using SISCI API☆17Apr 8, 2018Updated 7 years ago
- Basic data analytics on Facebook messages☆15Jul 5, 2020Updated 5 years ago
- An example to implement PBC SCF☆14Jul 10, 2018Updated 7 years ago
- NVIDIA DPU OPs collection☆15Mar 6, 2023Updated 3 years ago
- Julia wrapper for TBLIS tensor contraction library.☆18Nov 14, 2024Updated last year
- NBConverter stuff to convert Jupyter Notebooks in to Julia Literate.jl files (This is the opposite of Literate.jl)☆23May 11, 2020Updated 5 years ago
- Create Julia packages from Notebooks☆29Jul 30, 2022Updated 3 years ago
- Read audio with FFmpeg into NumPy/PyTorch via ctypes (standard library module)☆11Aug 12, 2020Updated 5 years ago
- performance engineering☆30Jul 11, 2024Updated last year
- JIT-compiled GPU kernels for quantum chemistry☆31Jan 30, 2026Updated last month
- Tensor Algebra for many-body methods☆19Feb 3, 2026Updated last month
- A GPU FP32 computation method with Tensor Cores.☆26Dec 8, 2025Updated 3 months ago
- Julia package for quantum circuit simulation using tensor networks☆40Feb 15, 2022Updated 4 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Oct 9, 2022Updated 3 years ago
- Large Graph Convolutional Network Training with GPU-Oriented Data Communication Architecture (accepted by PVLDB)☆44Jul 1, 2023Updated 2 years ago
- Static analysis framework for analyzing programs written in TVM's Relay IR.☆29Oct 31, 2019Updated 6 years ago
- Traits for julia iterators☆11Aug 22, 2020Updated 5 years ago
- Distributed-memory, arbitrary-precision, dense and sparse-direct linear algebra, conic optimization, and lattice reduction☆71Mar 17, 2025Updated last year
- Adds a macro (`@api`) to help manage a consistent API between multiple packages☆10May 18, 2021Updated 4 years ago
- A Julia IO type that facilitates width-limited printing☆12Mar 21, 2023Updated 3 years ago
- Software package to handle the many-fermionic operator☆16Jul 30, 2013Updated 12 years ago
- Companion repository to "Prompt Compression and Contrastive Conditioning for Controllability and Toxicity Reduction in Language Models"☆14May 31, 2023Updated 2 years ago
- ☆71Oct 6, 2025Updated 5 months ago
- A minimal block sparse symmetric and fermionic tensor python library☆31Updated this week
- ☆42Dec 10, 2024Updated last year
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Jan 16, 2022Updated 4 years ago
- Interactive plotting of millions of data points☆11Mar 31, 2022Updated 3 years ago
- ☆10Jul 12, 2017Updated 8 years ago
- A pure, low-level tensor program representation enabling tensor program optimization via program rewriting. See the web demo at https://g…☆73May 30, 2025Updated 9 months ago
- Functional programming utilities implemented in pure Lua.☆14Oct 30, 2024Updated last year