Alternatives and similar repositories for benchmark:

Users that are interested in benchmark are comparing it to the libraries listed below

• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆501Updated 3 years ago
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆773Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,032Updated 10 months ago
• pytorch / PiPPy
  Pipeline Parallelism for PyTorch
  ☆753Updated 6 months ago
• pytorch / TensorRT
  PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
  ☆2,688Updated this week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,264Updated this week
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆832Updated 7 months ago
• ryujaehun / pytorch-gpu-benchmark
  Using the famous cnn model in Pytorch, we run benchmarks on various gpu.
  ☆229Updated 8 months ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆952Updated last week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,269Updated last month
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…
  ☆2,233Updated this week
• pytorch / xla
  Enabling PyTorch on XLA Devices (e.g. Google TPU)
  ☆2,535Updated this week
• pytorch / builder
  Continuous builder and binary build scripts for pytorch
  ☆349Updated last week
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆513Updated 6 years ago
• pytorch / extension-cpp
  C++ extensions in PyTorch
  ☆1,068Updated last month
• lambdal / deeplearning-benchmark
  Benchmark Suite for Deep Learning
  ☆259Updated last week
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆340Updated last week
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆366Updated 6 years ago
• NVIDIA / nccl-tests
  NCCL Tests
  ☆1,011Updated this week
• pytorch / torchx
  TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…
  ☆346Updated last week
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,168Updated this week
• mlcommons / inference
  Reference implementations of MLPerf™ inference benchmarks
  ☆1,320Updated this week
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆298Updated 3 years ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆452Updated 5 years ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆546Updated 4 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,045Updated 6 months ago
• facebookincubator / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,266Updated this week
• pytorch / torcheval
  A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…
  ☆227Updated last month
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆449Updated this week
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆154Updated 2 months ago