A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
☆605Aug 14, 2026Updated 2 months ago
Alternatives and similar repositories for attorch
Users that are interested in attorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Cataloging released Triton kernels.☆314Sep 9, 2025Updated last year
- Experiment of using Tangent to autodiff triton☆81Jan 22, 2024Updated 2 years ago
- Puzzles for learning Triton☆2,656Apr 1, 2026Updated 7 months ago
- Applied AI experiments and examples for PyTorch☆322Aug 22, 2025Updated last year
- Tile primitives for speedy kernels☆3,790Sep 12, 2026Updated last month
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Efficient Triton Kernels for LLM Training☆6,683Updated this week
- Triton-based implementation of Sparse Mixture of Experts.☆281Oct 3, 2025Updated last year
- Fast low-bit matmul kernels in Triton☆491Oct 1, 2026Updated last month
- A collection of memory efficient attention operators implemented in the Triton language.☆327Updated this week
- ☆365Updated this week
- Collection of kernels written in Triton language☆200Jan 27, 2026Updated 9 months ago
- extensible collectives library in triton☆98Mar 31, 2025Updated last year
- Ring attention implementation with flash attention☆1,065Sep 10, 2025Updated last year
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆375Updated this week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Helpful tools and examples for working with flex-attention☆1,263Updated this week
- ☆119Jul 28, 2026Updated 3 months ago
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,584Jan 28, 2026Updated 9 months ago
- A Quirky Assortment of CuTe Kernels☆1,189Updated this week
- A PyTorch native platform for training generative AI models☆5,871Updated this week
- Shared Middle-Layer for Triton Compilation☆347Oct 15, 2026Updated 2 weeks ago
- UNet diffusion model in pure CUDA☆669Jun 28, 2024Updated 2 years ago
- Transformers components but in Triton☆34May 9, 2025Updated last year
- ☆11Sep 30, 2015Updated 11 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Accelerated First Order Parallel Associative Scan☆197Jan 7, 2026Updated 9 months ago
- ☆107Sep 9, 2024Updated 2 years ago
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,587Updated this week
- PyTorch native quantization for training and inference☆3,018Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆977Updated this week
- 🚀 Efficient implementations for emerging model architectures☆6,014Updated this week
- Schedule-Free Optimization in PyTorch☆2,333Jul 28, 2026Updated 3 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,336Aug 26, 2025Updated last year
- Development repository for the Triton language and compiler☆20,512Updated this week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Framework to reduce autotune overhead to zero for well known deployments.☆101Sep 19, 2025Updated last year
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,267Aug 22, 2025Updated last year
- Annotated version of the Mamba paper☆504Feb 27, 2024Updated 2 years ago
- flash attention tutorial written in python, triton, cuda, cutlass☆538Oct 22, 2026Updated last week
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated 2 years ago
- A dict[str, torch.Tensor] that behaves like one tensor, and TensorClass: typed, batched dataclasses that compile.☆1,042Updated this week
- ☆112Aug 26, 2024Updated 2 years ago