tspeterkim / mixed-precision-from-scratch

Related projects: ⓘ

• cuda-mode / profiling-cuda-in-torch
  ☆124Updated 7 months ago
• cuda-mode / triton-index
  Cataloging released Triton kernels.
  ☆111Updated 3 weeks ago
• cuda-mode / ring-attention
  ring-attention experiments
  ☆89Updated 5 months ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆88Updated 11 months ago
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA
  ☆48Updated this week
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆11Updated last month
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆152Updated 11 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆124Updated last week
• cloneofsimo / min-fsdp
  ☆68Updated 2 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆53Updated 9 months ago
• HanGuo97 / flute
  Fast Matrix Multiplications for Lookup Table-Quantized LLMs
  ☆156Updated this week
• neuralmagic / compressed-tensors
  A safetensors extension to efficiently store sparse quantized tensors on disk
  ☆26Updated this week
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆210Updated last month
• mgmalek / efficient_cross_entropy
  ☆66Updated 3 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆56Updated 9 months ago
• vdesai2014 / inference-optimization-blog-post
  ☆82Updated 6 months ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆123Updated last month
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆66Updated 3 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆150Updated last week
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆94Updated 2 weeks ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆77Updated 9 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆66Updated 7 months ago
• rasbt / pytorch-memory-optim
  This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…
  ☆84Updated last year
• AnswerDotAI / cold-compress
  Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…
  ☆73Updated last month
• HanGuo97 / lq-lora
  ☆117Updated 7 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆48Updated 2 weeks ago
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆62Updated 2 weeks ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆62Updated this week
• gnovack / distributed-training-and-deepspeed
  ☆13Updated last year
• mag- / gpu_benchmark
  Gpu benchmark
  ☆35Updated 2 weeks ago