Alternatives and similar repositories for nimble:

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

• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆253Updated 2 years ago
• AlibabaPAI / DAPPLE
  An Efficient Pipelined Data Parallel Approach for Training Large Model
  ☆74Updated 4 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆130Updated 3 years ago
• mit-han-lab / inter-operator-scheduler
  [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration
  ☆197Updated 2 years ago
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆835Updated 7 months ago
• snuspl / parallax
  A Tool for Automatic Parallelization of Deep Learning Training in Distributed Multi-GPU Environments.
  ☆130Updated 3 years ago
• cmu-catalyst / collage
  System for automated integration of deep learning backends.
  ☆48Updated 2 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆504Updated 3 years ago
• ConstantPark / DL_Compiler
  Study Group of Deep Learning Compiler
  ☆156Updated 2 years ago
• NVIDIA / nvtx-plugins
  Python bindings for NVTX
  ☆66Updated last year
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆342Updated this week
• msr-fiddle / pipedream
  ☆388Updated 2 years ago
• spcl / substation
  Research and development for optimizing transformers
  ☆125Updated 4 years ago
• andersy005 / tvm-in-action
  TVM stack: exploring the incredible explosion of deep-learning frameworks and how to bring them together
  ☆64Updated 6 years ago
• awslabs / raf
  ☆141Updated last month
• jiazhihao / metaflow_sysml19
  Repository for SysML19 Artifacts Evaluation
  ☆53Updated 6 years ago
• Tiiiger / QPyTorch
  Low Precision Arithmetic Simulation in PyTorch
  ☆272Updated 9 months ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆119Updated 2 years ago
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆69Updated 3 years ago
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆63Updated 2 years ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆298Updated 3 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆133Updated last year
• tlc-pack / TLCBench
  Benchmark scripts for TVM
  ☆73Updated 2 years ago
• petuum / autodist
  Simple Distributed Deep Learning on TensorFlow
  ☆134Updated 2 years ago
• jwfromm / Riptide
  Simple Training and Deployment of Fast End-to-End Binary Networks
  ☆158Updated 3 years ago
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆703Updated 2 years ago
• microsoft / varuna
  ☆246Updated 7 months ago
• pku-liang / FlexTensor
  Automatic Schedule Exploration and Optimization Framework for Tensor Computations
  ☆175Updated 2 years ago
• cli99 / flops-profiler
  pytorch-profiler
  ☆51Updated last year
• google / nccl-fastsocket
  NCCL Fast Socket is a transport layer plugin to improve NCCL collective communication performance on Google Cloud.
  ☆116Updated last year