dorpxam / einops-cpp
C++17 implementation of einops for libtorch - clear and reliable tensor manipulations with einstein-like notation
☆10Updated last year
Alternatives and similar repositories for einops-cpp:
Users that are interested in einops-cpp are comparing it to the libraries listed below
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆53Updated this week
- extensible collectives library in triton☆76Updated 3 months ago
- ☆25Updated last year
- ☆15Updated 3 months ago
- Sparsity support for PyTorch☆33Updated last month
- ☆154Updated 7 months ago
- Experiment of using Tangent to autodiff triton☆74Updated 11 months ago
- Accelerated First Order Parallel Associative Scan☆169Updated 4 months ago
- ☆64Updated 2 months ago
- FlexAttention w/ FlashAttention3 Support☆27Updated 3 months ago
- A library for unit scaling in PyTorch☆118Updated last month
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 10 months ago
- A parallel framework for training deep neural networks☆49Updated this week
- Collection of kernels written in Triton language☆90Updated 2 months ago
- The simplest but fast implementation of matrix multiplication in CUDA.☆34Updated 5 months ago
- ☆74Updated last year
- ☆27Updated 3 weeks ago
- A Data-Centric Compiler for Machine Learning☆82Updated last year
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆91Updated this week
- ☆34Updated last month
- ☆21Updated 2 months ago
- ☆50Updated 5 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆75Updated this week
- End to End steps for adding custom ops in PyTorch.☆19Updated 4 years ago
- Framework to reduce autotune overhead to zero for well known deployments.☆57Updated 2 months ago
- ☆16Updated last month
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆238Updated this week
- Fast Hadamard transform in CUDA, with a PyTorch interface☆133Updated 7 months ago
- JAX bindings for Flash Attention v2☆83Updated 6 months ago
- Material for the SC22 Deep Learning at Scale Tutorial☆39Updated last year