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
- OSLO: Open Source for Large-scale Optimization☆175Updated 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☆118Updated last year
- ☆26Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆94Updated last year
- ring-attention experiments☆147Updated 9 months ago
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆35Updated last year
- ☆45Updated last year
- Load compute kernels from the Hub☆233Updated this week
- some common Huggingface transformers in maximal update parametrization (µP)☆82Updated 3 years ago
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Updated 2 years ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆61Updated 10 months ago
- ☆114Updated last year
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆83Updated 3 weeks ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆160Updated 3 months ago
- Easy and Efficient Quantization for Transformers☆198Updated last month
- This is a fork of SGLang for hip-attention integration. Please refer to hip-attention for detail.☆15Updated this week
- BCQ tutorial for transformers☆17Updated 2 years ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆66Updated 8 months ago
- Mixed precision training from scratch with Tensors and CUDA☆24Updated last year
- Pytorch/XLA SPMD Test code in Google TPU☆23Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆80Updated 11 months ago
- ☆81Updated last year
- ☆103Updated 2 years ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆123Updated 8 months ago
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆95Updated last year
- Training-free Post-training Efficient Sub-quadratic Complexity Attention. Implemented with OpenAI Triton.☆142Updated this week