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 2 years ago
- Pytorch/XLA SPMD Test code in Google TPU☆23Updated last year
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆118Updated last year
- Mixed precision training from scratch with Tensors and CUDA☆28Updated last year
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆91Updated 4 months ago
- Flexibly track outputs and grad-outputs of torch.nn.Module.☆13Updated 2 years ago
- Automatic differentiation for Triton Kernels☆30Updated 3 months ago
- ☆27Updated last year
- OSLO: Open Source for Large-scale Optimization☆174Updated 2 years ago
- ☆103Updated 2 years ago
- A hackable, simple, and reseach-friendly GRPO Training Framework with high speed weight synchronization in a multinode environment.☆31Updated 2 months ago
- ring-attention experiments☆155Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated last year
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Updated 2 years ago
- Collection of kernels written in Triton language☆167Updated 7 months ago
- Experiment of using Tangent to autodiff triton☆80Updated last year
- Triton-based Symmetric Memory operators and examples☆63Updated last month
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆95Updated last year
- Lightning support for Intel Habana accelerators.☆26Updated 3 months ago
- ☆14Updated 8 months ago
- Transformers components but in Triton☆34Updated 6 months ago
- ☆83Updated last year
- some common Huggingface transformers in maximal update parametrization (µP)☆86Updated 3 years ago
- This repository contains the experimental PyTorch native float8 training UX☆225Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆103Updated last month
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆98Updated 2 years ago
- ☆66Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆85Updated last year
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆36Updated last year
- ☆47Updated last year