Alternatives and similar repositories for triton-flash-attention:

Users that are interested in triton-flash-attention are comparing it to the libraries listed below

• gpu-mode / profiling-cuda-in-torch
  ☆141Updated last year
• gpu-mode / ring-attention
  ring-attention experiments
  ☆123Updated 3 months ago
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆263Updated last week
• huggingface / picotron_tutorial
  ☆132Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆175Updated this week
• thevasudevgupta / gpt-triton
  Triton implementation of GPT/LLAMA
  ☆16Updated 5 months ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆164Updated last month
• hkproj / quantization-notes
  Notes on quantization in neural networks
  ☆68Updated last year
• NVIDIA / kvpress
  LLM KV cache compression made easy
  ☆384Updated this week
• facebookresearch / LayerSkip
  Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024
  ☆270Updated this week
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆221Updated this week
• NVIDIA / Star-Attention
  Efficient LLM Inference over Long Sequences
  ☆356Updated this week
• rasbt / dora-from-scratch
  LoRA and DoRA from Scratch Implementations
  ☆195Updated 11 months ago
• hkproj / pytorch-transformer-distributed
  Distributed training (multi-node) of a Transformer model
  ☆53Updated 10 months ago
• hkproj / pytorch-llama
  LLaMA 2 implemented from scratch in PyTorch
  ☆292Updated last year
• hkproj / pytorch-lora
  LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch
  ☆94Updated last year
• tspeterkim / mixed-precision-from-scratch
  Mixed precision training from scratch with Tensors and CUDA
  ☆21Updated 9 months ago
• Maharshi-Pandya / cudacodes
  Learnings and programs related to CUDA
  ☆229Updated this week
• NVlabs / hymba
  ☆165Updated 2 months ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆513Updated this week
• huggingface / picotron
  Minimalistic 4D-parallelism distributed training framework for education purpose
  ☆717Updated this week
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆348Updated 3 weeks ago
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆169Updated last year
• mengxiayu / LLMSuperWeight
  Code for studying the super weight in LLM
  ☆79Updated 2 months ago
• ThinamXx / Meta-llama
  Complete implementation of Llama2 with/without KV cache & inference 🚀
  ☆47Updated 8 months ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆117Updated last year
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆223Updated this week