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
  ☆257Updated 2 years ago
• mit-han-lab / inter-operator-scheduler
  [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration
  ☆200Updated 3 years ago
• ConstantPark / DL_Compiler
  Study Group of Deep Learning Compiler
  ☆159Updated 2 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆131Updated 3 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆503Updated 3 years ago
• snuspl / parallax
  A Tool for Automatic Parallelization of Deep Learning Training in Distributed Multi-GPU Environments.
  ☆130Updated 3 years ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆379Updated this week
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆842Updated 10 months ago
• NVIDIA / nvtx-plugins
  Python bindings for NVTX
  ☆66Updated last year
• spcl / substation
  Research and development for optimizing transformers
  ☆126Updated 4 years ago
• msr-fiddle / pipedream
  ☆390Updated 2 years ago
• awslabs / raf
  ☆143Updated 4 months ago
• AlibabaPAI / DAPPLE
  An Efficient Pipelined Data Parallel Approach for Training Large Model
  ☆76Updated 4 years ago
• andersy005 / tvm-in-action
  TVM stack: exploring the incredible explosion of deep-learning frameworks and how to bring them together
  ☆64Updated 7 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 …
  ☆65Updated 3 years ago
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆69Updated 4 years ago
• yalue / cuda_scheduling_examiner_mirror
  A tool for examining GPU scheduling behavior.
  ☆83Updated 9 months ago
• cmu-catalyst / collage
  System for automated integration of deep learning backends.
  ☆48Updated 2 years ago
• awslabs / lorien
  ☆43Updated last year
• swsnu / aisys2023
  ☆100Updated last year
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆714Updated 2 years ago
• SymbioticLab / Salus
  Fine-grained GPU sharing primitives
  ☆141Updated 5 years ago
• microsoft / varuna
  ☆249Updated 10 months ago
• utsaslab / MONeT
  MONeT framework for reducing memory consumption of DNN training
  ☆173Updated 4 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆132Updated last year
• sakjain92 / Fractional-GPUs
  Splits single Nvidia GPU into multiple partitions with complete compute and memory isolation (wrt to performace) between the partitions
  ☆159Updated 6 years ago
• marsupialtail / sparsednn
  Fast sparse deep learning on CPUs
  ☆53Updated 2 years ago
• swsnu / aisys2021
  ☆15Updated 3 years ago
• petuum / autodist
  Simple Distributed Deep Learning on TensorFlow
  ☆134Updated 2 years ago
• google / nccl-fastsocket
  NCCL Fast Socket is a transport layer plugin to improve NCCL collective communication performance on Google Cloud.
  ☆116Updated last year