microsoft / tensorflow-rematerialization

Related projects: ⓘ

• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆70Updated 2 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 years ago
• ray-project / distml
  Distributed ML Optimizer
  ☆31Updated 3 years ago
• microsoft / torchy
  A tracing JIT for PyTorch
  ☆18Updated 2 years ago
• facebookresearch / DLRM-FlexFlow
  Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…
  ☆28Updated 2 years ago
• zdevito / custom_loader
  ☆12Updated 3 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• iree-org / iree-jax
  ☆43Updated last month
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆61Updated 2 years ago
• microsoft / DGT
  Learning Accurate Decision Trees with Bandit Feedback via Quantized Gradient Descent
  ☆14Updated 2 years ago
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆19Updated 2 weeks 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
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆24Updated 8 months ago
• sail-sg / hloenv
  an environment based on XLA for deep learning compiler optimization research.
  ☆22Updated last year
• HazyResearch / mongoose
  A Learnable LSH Framework for Efficient NN Training
  ☆29Updated 3 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆127Updated 2 years ago
• Jokeren / triton-samples
  ☆22Updated last week
• hogepodge / tvm-docker
  A basic Docker-based installation of TVM
  ☆12Updated 2 years ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• uwsampl / relay-aot
  An experimental ahead of time compiler for Relay.
  ☆51Updated 4 years ago
• xldrx / tensorflow-tracer
  Runtime Tracing Library for TensorFlow
  ☆42Updated 5 years ago
• manishucsd / py-codegen
  ☆14Updated 4 months ago
• microsoft / DeepSpeed-Kernels
  ☆50Updated 3 months ago
• jiazhihao / sosp19ae
  Artifacts for SOSP'19 paper Optimizing Deep Learning Computation with Automatic Generation of Graph Substitutions
  ☆21Updated 2 years ago
• srush / transformers-bet
  ☆12Updated 2 years ago
• zhisbug / ray-scalable-ml-design
  Some microbenchmarks and design docs before commencement
  ☆12Updated 3 years ago
• microsoft / Analysis-Framework-for-TVM
  Static analysis framework for analyzing programs written in TVM's Relay IR.
  ☆27Updated 4 years ago
• google-research / structured-additive-IR
  ☆50Updated this week
• ezyang / stride-visualizer
  Stride visualizations
  ☆36Updated 6 years ago