tspeterkim / mixed-precision-from-scratchLinks
Mixed precision training from scratch with Tensors and CUDA
☆28Updated 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:
- ring-attention experiments☆165Updated last year
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆92Updated 6 months ago
- Cataloging released Triton kernels.☆291Updated 4 months ago
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- ☆178Updated last year
- a minimal cache manager for PagedAttention, on top of llama3.☆133Updated last year
- Collection of kernels written in Triton language☆175Updated 9 months ago
- ☆132Updated 8 months ago
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆119Updated last year
- Code for studying the super weight in LLM☆120Updated last year
- Load compute kernels from the Hub☆381Updated last week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆219Updated this week
- ☆92Updated last year
- A bunch of kernels that might make stuff slower 😉☆75Updated last week
- ☆117Updated 3 weeks ago
- Learn CUDA with PyTorch☆185Updated last week
- Fast low-bit matmul kernels in Triton☆424Updated this week
- Triton-based implementation of Sparse Mixture of Experts.☆263Updated 3 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆93Updated last year
- Applied AI experiments and examples for PyTorch☆314Updated 5 months ago
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆250Updated 8 months ago
- ☆277Updated this week
- Triton-based Symmetric Memory operators and examples☆80Updated 2 weeks ago
- ☆27Updated 2 years ago
- TPU inference for vLLM, with unified JAX and PyTorch support.☆223Updated this week
- The evaluation framework for training-free sparse attention in LLMs☆114Updated this week
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆148Updated last year
- Ship correct and fast LLM kernels to PyTorch☆139Updated 2 weeks ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆131Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆195Updated 7 months ago