jiazhihao / TASO

Related projects: ⓘ

• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆952Updated this week
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆640Updated 2 years ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆452Updated 4 years ago
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆742Updated this week
• tensorflow / mlir-hlo
  ☆392Updated this week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,169Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆890Updated last week
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆796Updated 3 weeks ago
• tlc-pack / relax
  ☆193Updated last year
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,301Updated this week
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆753Updated last month
• tpoisonooo / how-to-optimize-gemm
  row-major matmul optimization
  ☆584Updated last year
• msr-fiddle / pipedream
  ☆375Updated last year
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆380Updated this week
• mit-han-lab / inter-operator-scheduler
  [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration
  ☆191Updated 2 years ago
• tqchen / ffi-navigator
  ☆226Updated last year
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆646Updated 3 weeks ago
• bytedance / byteir
  A model compilation solution for various hardware
  ☆357Updated this week
• alibaba / heterogeneity-aware-lowering-and-optimization
  heterogeneity-aware-lowering-and-optimization
  ☆249Updated 7 months ago
• cloudcores / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆388Updated last year
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆692Updated this week
• bytedance / effective_transformer
  Running BERT without Padding
  ☆455Updated 2 years ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆247Updated last year
• facebookincubator / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,192Updated 2 months ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆994Updated 5 months ago
• pigirons / cpufp
  A CPU tool for benchmarking the peak of floating points
  ☆472Updated this week
• awslabs / raf
  ☆141Updated last year
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆240Updated 3 years ago
• baidu-research / baidu-allreduce
  ☆562Updated 6 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆493Updated 3 years ago