Alternatives and similar repositories for out-of-the-box-fp8-training:

Users that are interested in out-of-the-box-fp8-training are comparing it to the libraries listed below

• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆75Updated last year
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆122Updated 2 months ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated last year
• cloneofsimo / min-fsdp
  ☆75Updated 7 months ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆46Updated 2 months ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆46Updated last year
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆221Updated 6 months ago
• stanford-futuredata / stk
  ☆100Updated 5 months ago
• mgmalek / efficient_cross_entropy
  ☆91Updated 8 months ago
• UmerHA / triton_util
  Make triton easier
  ☆44Updated 8 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆83Updated 4 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated 11 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated last month
• lianakoleva / no-libtorch-compile
  ☆21Updated 3 months ago
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆67Updated 8 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 2 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆17Updated 7 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆89Updated this week
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆46Updated this week
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆75Updated 2 months ago
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆85Updated 7 months ago
• cat-state / tinypar
  ☆20Updated last year
• PiotrNawrot / nano-sparse-attention
  The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.
  ☆57Updated 3 weeks ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆157Updated last year
• google-deepmind / asyncdiloco
  ☆42Updated last year
• siyan-zhao / prepacking
  The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"
  ☆59Updated 4 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• IST-DASLab / SparseFinetuning
  Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry
  ☆40Updated last year