Alternatives and similar repositories for pytorch-benchmarks

Users that are interested in pytorch-benchmarks are comparing it to the libraries listed below

• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆62Updated 4 months ago
• HabanaAI / Megatron-DeepSpeed
  Intel Gaudi's Megatron DeepSpeed Large Language Models for training
  ☆13Updated 6 months ago
• FrancescoSaverioZuppichini / pytorch-2.0-benchmark
  Benchmarking PyTorch 2.0 different models
  ☆21Updated 2 years ago
• groq / mlagility
  Machine Learning Agility (MLAgility) benchmark and benchmarking tools
  ☆39Updated 2 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…
  ☆92Updated 2 years ago
• Michaelvll / llm-ie-benchmarks
  A collection of reproducible inference engine benchmarks
  ☆32Updated 2 months ago
• 7shoe / AdaParse
  Adaptive Parallel PDF Parsing and Resource Scaling Engine
  ☆48Updated last month
• cray-lm / cray-lm
  Cray-LM unified training and inference stack.
  ☆22Updated 5 months ago
• foundation-model-stack / bamba
  Train, tune, and infer Bamba model
  ☆130Updated last month
• EmbeddedLLM / vllm
  vLLM: A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆87Updated this week
• OpenGPTX / lm-evaluation-harness
  A framework for few-shot evaluation of autoregressive language models.
  ☆11Updated this week
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆46Updated this week
• tanyuqian / redco
  NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
  ☆66Updated 7 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• at-aaims / forge
  ☆15Updated 2 months ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated last year
• spcl / sten
  Sparsity support for PyTorch
  ☆35Updated 3 months ago
• graphcore-research / unit-scaling-demo
  Unit Scaling demo and experimentation code
  ☆16Updated last year
• rapidsai / cugraph-gnn
  ☆47Updated last week
• fishmingyu / GeoT
  GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU
  ☆23Updated 3 months ago
• facebookexperimental / protoquant
  Prototype routines for GPU quantization written using PyTorch.
  ☆21Updated 5 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆19Updated last year
• MiuLab / LLM-Eval
  ☆15Updated last year
• abacusai / gh200-llm
  Docker image NVIDIA GH200 machines - optimized for vllm serving and hf trainer finetuning
  ☆46Updated 4 months ago
• mlcommons / logging
  MLPerf™ logging library
  ☆37Updated 3 months ago
• HazyResearch / train-tk
  train with kittens!
  ☆61Updated 8 months ago
• deepspeedai / DeepSpeed-Kernels
  ☆74Updated 3 months ago
• sambanova / tutorials
  ☆12Updated last year
• determined-ai / determined-examples
  Example ML projects that use the Determined library.
  ☆32Updated 10 months ago