FrancescoSaverioZuppichini / pytorch-2.0-benchmark

Related projects ⓘ

Alternatives and complementary repositories for pytorch-2.0-benchmark

• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆59Updated 7 months ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• 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
• Doraemonzzz / Awesome-Triton-Resources
  Awesome Triton Resources
  ☆18Updated 3 weeks ago
• rlin27 / DeBut
  Codes of the paper Deformable Butterfly: A Highly Structured and Sparse Linear Transform.
  ☆12Updated 3 years ago
• lianakoleva / no-libtorch-compile
  ☆17Updated 2 weeks ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆20Updated 7 months ago
• brightlaboratory / polydl
  ☆11Updated 3 years ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆15Updated 3 months ago
• davisyoshida / abnormal-floats
  Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)
  ☆18Updated last year
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• lucidrains / rela-transformer
  Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012
  ☆49Updated 2 years ago
• Aleph-Alpha / NeurIPS-WANT-submission-efficient-parallelization-layouts
  ☆22Updated 10 months ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• facebookexperimental / protoquant
  Prototype routines for GPU quantization written using PyTorch.
  ☆19Updated last week
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated last month
• HabanaAI / Megatron-DeepSpeed
  Intel Gaudi's Megatron DeepSpeed Large Language Models for training
  ☆13Updated 3 weeks ago
• ptillet / triton-llvm-releases
  ☆20Updated last year
• BBuf / flash-rwkv
  ☆29Updated 5 months ago
• IST-DASLab / QIGen
  Repository for CPU Kernel Generation for LLM Inference
  ☆24Updated last year
• gau-nernst / quantized-training
  Explore training for quantized models
  ☆10Updated 2 weeks ago
• juntang-zhuang / ACProp-Optimizer
  Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)
  ☆15Updated 3 years ago
• tridao / flash-attention-wheels
  ☆41Updated 11 months ago
• tanyuqian / redco
  NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
  ☆60Updated 2 weeks ago
• buttercutter / Mamba_SSM
  A simple implementation of [Mamba: Linear-Time Sequence Modeling with Selective State Spaces](https://arxiv.org/abs/2312.00752)
  ☆19Updated 9 months ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• Itamarzimm / UnifiedImplicitAttnRepr
  ☆32Updated 5 months ago