unixpickle / learn-ptx

Related projects ⓘ

Alternatives and complementary repositories for learn-ptx

• Deep-Learning-Profiling-Tools / triton-viz
  ☆140Updated this week
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆132Updated 2 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆133Updated 9 months ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆159Updated last week
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆133Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆61Updated last month
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆95Updated last year
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆211Updated 3 months ago
• HanGuo97 / flute
  Fast Matrix Multiplications for Lookup Table-Quantized LLMs
  ☆183Updated last month
• ColfaxResearch / cutlass-kernels
  ☆162Updated 3 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆63Updated last week
• gpu-mode / ring-attention
  ring-attention experiments
  ☆95Updated 3 weeks ago
• vdesai2014 / inference-optimization-blog-post
  ☆82Updated 8 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆87Updated 3 months ago
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆139Updated 3 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆71Updated 9 months ago
• hyhieu / easy_pybind
  ☆32Updated 5 months ago
• triton-lang / kernels
  ☆43Updated this week
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆35Updated 3 months ago
• neuralmagic / compressed-tensors
  A safetensors extension to efficiently store sparse quantized tensors on disk
  ☆46Updated this week
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆66Updated 5 months ago
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆34Updated 5 months ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆146Updated this week
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆86Updated 6 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆156Updated last year
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆107Updated 5 months ago
• stanford-futuredata / stk
  ☆88Updated 2 months ago
• cloneofsimo / min-fsdp
  ☆72Updated 4 months ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆30Updated 3 months ago
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆157Updated last year