tspeterkim / mixed-precision-from-scratchLinks
Mixed precision training from scratch with Tensors and CUDA
☆27Updated last year
Alternatives and similar repositories for mixed-precision-from-scratch
Users that are interested in mixed-precision-from-scratch are comparing it to the libraries listed below
Sorting:
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆89Updated 2 months ago
- ☆173Updated last year
- Learn CUDA with PyTorch☆85Updated 2 weeks ago
- The evaluation framework for training-free sparse attention in LLMs☆101Updated 3 months ago
- This repository contains the experimental PyTorch native float8 training UX☆223Updated last year
- ring-attention experiments☆153Updated 11 months ago
- Load compute kernels from the Hub☆299Updated this week
- ☆129Updated 4 months ago
- ☆120Updated last year
- Cataloging released Triton kernels.☆261Updated last month
- Triton-based implementation of Sparse Mixture of Experts.☆244Updated last week
- Collection of kernels written in Triton language☆157Updated 6 months ago
- Code for studying the super weight in LLM☆120Updated 10 months ago
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆114Updated last week
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆147Updated last year
- Experiment of using Tangent to autodiff triton☆80Updated last year
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆168Updated last year
- ☆143Updated 7 months ago
- Fast low-bit matmul kernels in Triton☆379Updated 2 weeks ago
- a minimal cache manager for PagedAttention, on top of llama3.☆123Updated last year
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆144Updated last year
- A bunch of kernels that might make stuff slower 😉☆61Updated this week
- ☆242Updated this week
- Applied AI experiments and examples for PyTorch☆296Updated last month
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆193Updated 3 months ago
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆108Updated 2 years ago
- ☆112Updated last year
- ☆89Updated last year
- ☆100Updated last month
- Example of applying CUDA graphs to LLaMA-v2☆12Updated 2 years ago