Jaykef / Triton-nanoGPTLinks
Custom triton kernels for training Karpathy's nanoGPT.
☆19Updated 8 months ago
Alternatives and similar repositories for Triton-nanoGPT
Users that are interested in Triton-nanoGPT are comparing it to the libraries listed below
Sorting:
- Minimal but scalable implementation of large language models in JAX☆35Updated 7 months ago
- An implementation of the Llama architecture, to instruct and delight☆21Updated 3 weeks ago
- ☆53Updated last year
- ☆78Updated 11 months ago
- ☆53Updated last year
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- ☆37Updated last year
- Code and Configs for Asynchronous RLHF: Faster and More Efficient RL for Language Models☆57Updated last month
- Simple and efficient pytorch-native transformer training and inference (batched)☆76Updated last year
- Collection of autoregressive model implementation☆85Updated last month
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆134Updated this week
- Experiment of using Tangent to autodiff triton☆79Updated last year
- ☆20Updated last year
- supporting pytorch FSDP for optimizers☆82Updated 6 months ago
- ☆14Updated 11 months ago
- ☆79Updated 10 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆66Updated 2 months ago
- ☆48Updated last year
- train with kittens!☆59Updated 7 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 weeks ago
- Using FlexAttention to compute attention with different masking patterns☆43Updated 9 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆127Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆38Updated last week
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆70Updated last week
- ☆19Updated last month
- Experiments on the impact of depth in transformers and SSMs.☆30Updated 7 months ago
- ☆12Updated 3 months ago
- JAX implementation of the Mistral 7b v0.2 model☆35Updated 11 months ago
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆51Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆68Updated 10 months ago