srush / triton-autodiffView external linksLinks
Experiment of using Tangent to autodiff triton
☆82Jan 22, 2024Updated 2 years ago
Alternatives and similar repositories for triton-autodiff
Users that are interested in triton-autodiff are comparing it to the libraries listed below
Sorting:
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- train with kittens!☆63Oct 25, 2024Updated last year
- Blog post☆17Feb 16, 2024Updated 2 years ago
- ☆19Dec 4, 2025Updated 2 months ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Jun 6, 2024Updated last year
- ☆20May 30, 2024Updated last year
- ☆51Jan 28, 2024Updated 2 years ago
- Efficient PScan implementation in PyTorch☆17Jan 2, 2024Updated 2 years ago
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- Example of applying CUDA graphs to LLaMA-v2☆12Aug 25, 2023Updated 2 years ago
- ☆124May 28, 2024Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆596Aug 12, 2025Updated 6 months ago
- ☆22Dec 15, 2023Updated 2 years ago
- ☆16Dec 19, 2024Updated last year
- ☆53Jan 30, 2024Updated 2 years ago
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆40Dec 2, 2023Updated 2 years ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Jun 18, 2024Updated last year
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆30Jan 28, 2026Updated 2 weeks ago
- ☆39Apr 5, 2024Updated last year
- ☆115Aug 26, 2024Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32May 25, 2024Updated last year
- Simple and efficient pytorch-native transformer training and inference (batched)☆79Apr 2, 2024Updated last year
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆23Oct 15, 2024Updated last year
- ☆20Oct 11, 2023Updated 2 years ago
- ☆288Updated this week
- ☆35Feb 26, 2024Updated last year
- ☆11Oct 11, 2023Updated 2 years ago
- ☆12Jan 29, 2021Updated 5 years ago
- Display tensors directly from GPU☆11Oct 12, 2025Updated 4 months ago
- PyTorch implementation for PaLM: A Hybrid Parser and Language Model.☆10Jan 7, 2020Updated 6 years ago
- Advanced Formal Language Theory (263-5352-00L; Frühjahr 2023)☆10Feb 21, 2023Updated 2 years ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11May 8, 2024Updated last year
- Annotated version of the Mamba paper☆496Feb 27, 2024Updated last year
- Accelerated First Order Parallel Associative Scan☆196Jan 7, 2026Updated last month
- Understand and test language model architectures on synthetic tasks.☆252Jan 12, 2026Updated last month
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated last year
- ☆13Dec 15, 2025Updated 2 months ago
- ☆44Nov 1, 2025Updated 3 months ago