zdevito / custom_loader

Related projects ⓘ

Alternatives and complementary repositories for custom_loader

• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago
• ray-project / ray_shuffling_data_loader
  A Ray-based data loader with per-epoch shuffling and configurable pipelining, for shuffling and loading training data for distributed tra…
  ☆18Updated last year
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆70Updated 2 years ago
• microsoft / torchy
  A tracing JIT for PyTorch
  ☆17Updated 2 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• openucx / torch-ucc
  pytorch ucc plugin
  ☆16Updated 3 years ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 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
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆25Updated 9 months ago
• uwsampl / relay-aot
  An experimental ahead of time compiler for Relay.
  ☆51Updated 4 years ago
• google / mlir-npcomp
  npcomp - An aspirational MLIR based numpy compiler
  ☆51Updated 4 years ago
• facebookresearch / DLRM-FlexFlow
  Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…
  ☆28Updated 3 years ago
• jansel / pytorch-jit-paritybench
  ☆36Updated 5 months ago
• mlcommons / logging
  MLPerf™ logging library
  ☆30Updated last week
• nod-ai / SRT
  Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …
  ☆107Updated this week
• NVIDIA / LDDL
  Distributed preprocessing and data loading for language datasets
  ☆39Updated 6 months ago
• iree-org / iree-jax
  ☆48Updated 3 months ago
• microsoft / tensorflow-rematerialization
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Updated 4 years ago
• rapidsai / ucx-py
  Python bindings for UCX
  ☆121Updated this week
• danyangz / lightning
  Lightning In-Memory Object Store
  ☆44Updated 2 years ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆146Updated this week
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆21Updated 3 weeks ago
• ray-project / distml
  Distributed ML Optimizer
  ☆31Updated 3 years ago
• stellaraccident / mlir-py-release
  ☆11Updated 3 years ago
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆19Updated 4 years ago
• NVIDIA / HMM_sample_code
  CUDA 12.2 HMM demos
  ☆17Updated 3 months ago
• zhisbug / ray-scalable-ml-design
  Some microbenchmarks and design docs before commencement
  ☆12Updated 3 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 years ago