Alternatives and similar repositories for kineto:

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

• pytorch / PiPPy
  Pipeline Parallelism for PyTorch
  ☆763Updated 7 months ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆366Updated last week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,040Updated last year
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆506Updated 3 years ago
• pytorch / torchx
  TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…
  ☆358Updated this week
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆933Updated this week
• NVIDIA / nccl-tests
  NCCL Tests
  ☆1,065Updated last month
• microsoft / Tutel
  Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4
  ☆800Updated this week
• thuml / depyf
  depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.
  ☆641Updated this week
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆836Updated 8 months ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆256Updated 2 years ago
• msr-fiddle / pipedream
  ☆388Updated 2 years ago
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…
  ☆2,365Updated this week
• gpuopenanalytics / pynvml
  Provide Python access to the NVML library for GPU diagnostics
  ☆231Updated 4 months ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆156Updated 4 months ago
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,299Updated this week
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆263Updated 3 years ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆978Updated last week
• albanD / subclass_zoo
  ☆163Updated 10 months ago
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆981Updated 7 months ago
• microsoft / msccl
  Microsoft Collective Communication Library
  ☆343Updated last year
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆698Updated last month
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆146Updated this week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆317Updated this week
• Azure / MS-AMP
  Microsoft Automatic Mixed Precision Library
  ☆593Updated 6 months ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆604Updated 11 months ago
• 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…
  ☆230Updated 3 months ago
• bytedance / effective_transformer
  Running BERT without Padding
  ☆471Updated 3 years ago
• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆382Updated last week
• tensorflow / mlir-hlo
  ☆410Updated this week