thevasudevgupta / gpt-triton
Triton implementation of GPT/LLAMA
☆16Updated 6 months ago
Alternatives and similar repositories for gpt-triton:
Users that are interested in gpt-triton are comparing it to the libraries listed below
- ☆136Updated 2 months ago
- ☆76Updated 8 months ago
- Prune transformer layers☆68Updated 9 months ago
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆51Updated 11 months ago
- ☆158Updated last month
- supporting pytorch FSDP for optimizers☆79Updated 3 months ago
- ☆151Updated last year
- Cataloging released Triton kernels.☆204Updated 2 months ago
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆68Updated 9 months ago
- Custom triton kernels for training Karpathy's nanoGPT.☆18Updated 5 months ago
- Learn CUDA with PyTorch☆19Updated last month
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆127Updated last year
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆58Updated last month
- ring-attention experiments☆127Updated 5 months ago
- Code for studying the super weight in LLM☆91Updated 3 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆100Updated 4 months ago
- Custom kernels in Triton language for accelerating LLMs☆18Updated 11 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆122Updated 11 months ago
- Collection of autoregressive model implementation☆83Updated last month
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"☆59Updated 5 months ago
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆47Updated 10 months ago
- ☆191Updated this week
- Collection of kernels written in Triton language☆114Updated last month
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆232Updated 2 weeks ago