facebookresearch / CompilerGym

Related projects: ⓘ

• google / ml-compiler-opt
  Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.
  ☆613Updated this week
• 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
• zwang4 / awesome-machine-learning-in-compilers
  Must read research papers and links to tools and datasets that are related to using machine learning for compilers and systems optimisati…
  ☆1,430Updated 2 months ago
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆380Updated this week
• j2kun / mlir-tutorial
  MLIR For Beginners tutorial
  ☆731Updated last month
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆742Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆994Updated 5 months ago
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆646Updated 3 weeks ago
• EnzymeAD / Enzyme
  High-performance automatic differentiation of LLVM and MLIR.
  ☆1,253Updated this week
• iree-org / iree
  A retargetable MLIR-based machine learning compiler and runtime toolkit.
  ☆2,559Updated this week
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆687Updated last year
• tensorflow / mlir-hlo
  ☆392Updated this week
• ChrisCummins / ProGraML
  A Graph-based Program Representation for Data Flow Analysis and Compiler Optimizations
  ☆296Updated 3 months ago
• dmlc / dlpack
  common in-memory tensor structure
  ☆890Updated last week
• facebookresearch / rlmeta
  RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.
  ☆284Updated last year
• google-deepmind / rlax
  ☆1,234Updated 3 months ago
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆365Updated last year
• llvm / Polygeist
  C/C++ frontend for MLIR. Also features polyhedral optimizations, parallel optimizations, and more!
  ☆466Updated 3 weeks ago
• google-deepmind / launchpad
  ☆309Updated last year
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,169Updated this week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,389Updated this week
• tensorflow / runtime
  A performant and modular runtime for TensorFlow
  ☆753Updated last month
• tensorflow / mlir
  "Multi-Level Intermediate Representation" Compiler Infrastructure
  ☆1,735Updated 3 years ago
• google-deepmind / reverb
  Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
  ☆704Updated last month
• google / paxml
  Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…
  ☆445Updated last week
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆2,577Updated this week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆692Updated this week
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,230Updated 5 months ago
• google / XNNPACK
  High-efficiency floating-point neural network inference operators for mobile, server, and Web
  ☆1,812Updated this week
• spcl / pymlir
  Python interface for MLIR - the Multi-Level Intermediate Representation
  ☆210Updated 3 months ago