ZixuanJiang / pre-rmsnorm-transformer
☆24Updated last year
Alternatives and similar repositories for pre-rmsnorm-transformer:
Users that are interested in pre-rmsnorm-transformer are comparing it to the libraries listed below
- Experiment of using Tangent to autodiff triton☆75Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 7 months ago
- train with kittens!☆53Updated 3 months ago
- ☆138Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆143Updated 8 months ago
- Fast and memory-efficient exact attention☆58Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆221Updated 6 months ago
- ☆21Updated 3 months ago
- extensible collectives library in triton☆83Updated 4 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆103Updated 2 months ago
- Make triton easier☆44Updated 8 months ago
- ☆201Updated 2 years ago
- Explore training for quantized models☆15Updated last month
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 11 months ago
- ML/DL Math and Method notes☆58Updated last year
- Collection of kernels written in Triton language☆105Updated this week
- PB-LLM: Partially Binarized Large Language Models☆151Updated last year
- ☆285Updated last week
- Sparsity support for PyTorch☆33Updated last week
- ShiftAddLLM: Accelerating Pretrained LLMs via Post-Training Multiplication-Less Reparameterization☆102Updated 4 months ago
- ☆77Updated last year
- ☆52Updated 4 months ago
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆296Updated last month
- FlashRNN - Fast RNN Kernels with I/O Awareness☆75Updated 2 months ago
- Cataloging released Triton kernels.☆168Updated last month
- Accelerated First Order Parallel Associative Scan☆171Updated 6 months ago
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆229Updated this week
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆57Updated 3 weeks ago
- ☆100Updated 5 months ago
- seqax = sequence modeling + JAX☆143Updated 7 months ago