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
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆87Updated last year
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆118Updated last year
- A hackable, simple, and reseach-friendly GRPO Training Framework with high speed weight synchronization in a multinode environment.☆31Updated 3 weeks ago
- Pytorch/XLA SPMD Test code in Google TPU☆23Updated last year
- ring-attention experiments☆152Updated 11 months ago
- OSLO: Open Source for Large-scale Optimization☆175Updated 2 years ago
- ☆27Updated last year
- ☆83Updated last year
- ☆119Updated last year
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆127Updated 9 months ago
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Updated 2 years ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆87Updated 2 months ago
- Mixed precision training from scratch with Tensors and CUDA☆27Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated 11 months ago
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆96Updated 2 years ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- ☆46Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆82Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆94Updated last year
- BCQ tutorial for transformers☆17Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆81Updated last year
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆161Updated 5 months ago
- Easy and Efficient Quantization for Transformers☆203Updated 2 months ago
- Flexibly track outputs and grad-outputs of torch.nn.Module.☆13Updated last year
- ☆103Updated 2 years ago
- Load compute kernels from the Hub☆287Updated this week
- some common Huggingface transformers in maximal update parametrization (µP)☆82Updated 3 years ago
- ☆112Updated last year
- ☆89Updated last year