Jokeren / triton-samples

Related projects ⓘ

Alternatives and complementary repositories for triton-samples

• cchan / tccl
  extensible collectives library in triton
  ☆72Updated last month
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• yifuwang / symm-mem-recipes
  ☆12Updated last month
• triton-lang / kernels
  ☆45Updated 2 weeks ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆60Updated 8 months ago
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆145Updated this week
• stanford-futuredata / stk
  ☆88Updated 2 months ago
• exists-forall / striped_attention
  ☆36Updated last year
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆134Updated 2 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆68Updated 3 weeks ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆185Updated last month
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆166Updated 3 weeks ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆16Updated 4 months ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆153Updated this week
• albanD / subclass_zoo
  ☆148Updated 5 months ago
• ColfaxResearch / cutlass-kernels
  ☆169Updated 4 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆105Updated 2 weeks ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆107Updated last year
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆211Updated 3 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated 3 weeks ago
• tspeterkim / paged-attention-minimal
  a minimal cache manager for PagedAttention, on top of llama3.
  ☆45Updated 2 months ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆165Updated this week
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 5 months ago
• tanyuqian / redco
  NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
  ☆61Updated last month
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆59Updated 6 years ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆97Updated last month
• pytorch-labs / superblock
  A block oriented training approach for inference time optimization.
  ☆30Updated 3 months ago