pytorch / elastic

Related projects ⓘ

Alternatives and complementary repositories for elastic

• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆493Updated 3 years ago
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆814Updated 3 months ago
• Santosh-Gupta / SpeedTorch
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆683Updated 4 years ago
• awwong1 / torchprof
  PyTorch layer-by-layer model profiler
  ☆608Updated 3 years ago
• kubeflow / pytorch-operator
  PyTorch on Kubernetes
  ☆306Updated 2 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…
  ☆332Updated 2 weeks ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆545Updated 3 years ago
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆648Updated 7 months ago
• uber / neuropod
  A uniform interface to run deep learning models from multiple frameworks
  ☆936Updated 10 months ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆453Updated 4 years ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆905Updated last month
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆727Updated this week
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆296Updated 3 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,018Updated 3 months ago
• tensorflow / mesh
  Mesh TensorFlow: Model Parallelism Made Easier
  ☆1,591Updated 11 months ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,010Updated 6 months ago
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆211Updated last year
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆696Updated last year
• msr-fiddle / pipedream
  ☆376Updated 2 years ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆247Updated last year
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆756Updated 3 weeks ago
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆253Updated last year
• openai / blocksparse
  Efficient GPU kernels for block-sparse matrix multiplication and convolution
  ☆1,025Updated last year
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,394Updated this week
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆391Updated 3 years ago
• facebookincubator / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,219Updated this week
• perone / euclidesdb
  A multi-model machine learning feature embedding database
  ☆631Updated 4 years ago
• tensorflow / custom-op
  Guide for building custom op for TensorFlow
  ☆378Updated last year
• snuspl / nimble
  Lightweight and Parallel Deep Learning Framework
  ☆263Updated last year
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,131Updated this week