microsoft / torchy

Related projects ⓘ

Alternatives and complementary repositories for torchy

• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆25Updated 9 months ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• iree-org / iree-llvm-sandbox
  A sandbox for quick iteration and experimentation on projects related to IREE, MLIR, and LLVM
  ☆54Updated last month
• uwsampl / relay-aot
  An experimental ahead of time compiler for Relay.
  ☆51Updated 4 years ago
• jiazhihao / sosp19ae
  Artifacts for SOSP'19 paper Optimizing Deep Learning Computation with Automatic Generation of Graph Substitutions
  ☆21Updated 2 years ago
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆21Updated 3 weeks ago
• chips-compilers-mlsys-21 / chips-compilers-mlsys-21.github.io
  ☆11Updated 3 years ago
• zdevito / custom_loader
  ☆12Updated 3 years ago
• microsoft / Analysis-Framework-for-TVM
  Static analysis framework for analyzing programs written in TVM's Relay IR.
  ☆27Updated 5 years ago
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆34Updated 5 months ago
• makslevental / nelli
  A lightweight, Pythonic, frontend for MLIR
  ☆79Updated last year
• 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
• openxla / shardy
  MLIR-based partitioning system
  ☆36Updated this week
• tlc-pack / tlcpack
  ☆23Updated 8 months ago
• google / mlir-npcomp
  npcomp - An aspirational MLIR based numpy compiler
  ☆51Updated 4 years ago
• jansel / pytorch-jit-paritybench
  ☆36Updated 5 months ago
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆44Updated 5 years ago
• microsoft / Accera
  Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research
  ☆101Updated last year
• microsoft / tensorflow-rematerialization
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Updated 4 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• feifeibear / PSTensor
  PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.
  ☆10Updated 2 years ago
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆19Updated 4 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 years ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• iml130 / mlir-emitc
  Conversions to MLIR EmitC
  ☆124Updated 2 months ago
• anony-sub / chameleon
  Chameleon: Adaptive Code Optimization for Expedited Deep Neural Network Compilation
  ☆26Updated 5 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• Bruce-Lee-LY / cuda_back2back_hgemm
  Use tensor core to calculate back-to-back HGEMM (half-precision general matrix multiplication) with MMA PTX instruction.
  ☆11Updated last year