daemyung / practice-tritonLinks
삼각형의 실전! Triton
☆16Updated last year
Alternatives and similar repositories for practice-triton
Users that are interested in practice-triton are comparing it to the libraries listed below
Sorting:
- A performance library for machine learning applications.☆184Updated last year
- Flexibly track outputs and grad-outputs of torch.nn.Module.☆13Updated last year
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆117Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆93Updated last year
- ring-attention experiments☆143Updated 9 months ago
- Load compute kernels from the Hub☆207Updated this week
- ☆81Updated last year
- BCQ tutorial for transformers☆17Updated 2 years ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 11 months ago
- ☆26Updated last year
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆85Updated last year
- Collection of kernels written in Triton language☆136Updated 3 months ago
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Updated 2 years ago
- ☆102Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆79Updated last year
- some common Huggingface transformers in maximal update parametrization (µP)☆81Updated 3 years ago
- Pytorch/XLA SPMD Test code in Google TPU☆23Updated last year
- Triton-based implementation of Sparse Mixture of Experts.☆225Updated 7 months ago
- Learn CUDA with PyTorch☆29Updated this week
- ☆88Updated last year
- ☆112Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆80Updated 10 months ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated 9 months ago
- Easy and Efficient Quantization for Transformers☆198Updated 3 weeks ago
- ☆106Updated 10 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆78Updated last month
- OSLO: Open Source for Large-scale Optimization☆175Updated last year
- ☆45Updated last year
- ☆13Updated 4 months ago
- Transformers components but in Triton☆34Updated 2 months ago