BaguaSys / bagua

Related projects ⓘ

Alternatives and complementary repositories for bagua

• PersiaML / PERSIA
  High performance distributed framework for training deep learning recommendation models based on PyTorch.
  ☆397Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,011Updated 7 months ago
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆734Updated this week
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆296Updated 3 years ago
• bytedance / effective_transformer
  Running BERT without Padding
  ☆460Updated 2 years ago
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆962Updated 2 months ago
• alibaba / EasyParallelLibrary
  Easy Parallel Library (EPL) is a general and efficient deep learning framework for distributed model training.
  ☆263Updated last year
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆816Updated this week
• volcengine / veGiantModel
  ☆209Updated last year
• Tencent / TurboTransformers
  a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.
  ☆1,484Updated last year
• petuum / adaptdl
  Resource-adaptive cluster scheduler for deep learning training.
  ☆426Updated last year
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆495Updated 3 years ago
• Tencent / PatrickStar
  PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.
  ☆747Updated last year
• msr-fiddle / pipedream
  ☆378Updated 2 years ago
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆643Updated 2 years ago
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆714Updated 2 years ago
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,210Updated this week
• facebookincubator / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,227Updated this week
• NVIDIA-Merlin / HugeCTR
  HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
  ☆946Updated last month
• pytorch / elastic
  PyTorch elastic training
  ☆730Updated 2 years ago
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆818Updated 3 months 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…
  ☆332Updated this week
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆692Updated last year
• microsoft / Tutel
  Tutel MoE: An Optimized Mixture-of-Experts Implementation
  ☆735Updated this week
• Oneflow-Inc / DLPerf
  DeepLearning Framework Performance Profiling Toolkit
  ☆277Updated 2 years ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆249Updated last year
• Oneflow-Inc / libai
  LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training
  ☆390Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆912Updated last month
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆649Updated 7 months ago
• Tencent / Forward
  A library for high performance deep learning inference on NVIDIA GPUs.
  ☆547Updated 2 years ago