Alternatives and similar repositories for CompilerGym:

Users that are interested in CompilerGym are comparing it to the libraries listed below

• google / ml-compiler-opt
  Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.
  ☆657Updated 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,499Updated last month
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆446Updated last week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,424Updated this week
• j2kun / mlir-tutorial
  MLIR For Beginners tutorial
  ☆905Updated 2 weeks ago
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆812Updated this week
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆696Updated 2 years ago
• tensorflow / mlir-hlo
  ☆406Updated this week
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆681Updated last week
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆366Updated 2 years ago
• iree-org / iree
  A retargetable MLIR-based machine learning compiler and runtime toolkit.
  ☆2,998Updated this week
• google-deepmind / launchpad
  ☆319Updated last year
• google-deepmind / rlax
  ☆1,288Updated 3 weeks ago
• llvm / Polygeist
  C/C++ frontend for MLIR. Also features polyhedral optimizations, parallel optimizations, and more!
  ☆511Updated 4 months ago
• EnzymeAD / Enzyme
  High-performance automatic differentiation of LLVM and MLIR.
  ☆1,338Updated this week
• google-deepmind / reverb
  Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
  ☆716Updated 3 weeks ago
• facebookresearch / rlmeta
  RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.
  ☆283Updated 2 years ago
• xdslproject / xdsl
  A Python Compiler Design Toolkit
  ☆312Updated this week
• openxla / community
  Stores documents and resources used by the OpenXLA developer community
  ☆117Updated 6 months ago
• spcl / pymlir
  Python interface for MLIR - the Multi-Level Intermediate Representation
  ☆240Updated 2 months ago
• tensorflow / mlir
  "Multi-Level Intermediate Representation" Compiler Infrastructure
  ☆1,737Updated 3 years ago
• NVlabs / cule
  CuLE: A CUDA port of the Atari Learning Environment (ALE)
  ☆233Updated 2 years ago
• intel / neuro-vectorizer
  NeuroVectorizer is a framework that uses deep reinforcement learning (RL) to predict optimal vectorization compiler pragmas for for loops…
  ☆92Updated 2 years ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆379Updated 3 weeks ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,029Updated 10 months ago
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆974Updated 5 months ago
• salesforce / warp-drive
  Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)
  ☆474Updated 6 months ago
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆2,966Updated this week
• google-deepmind / lab2d
  A customisable 2D platform for agent-based AI research
  ☆424Updated last year
• sail-sg / envpool
  C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.
  ☆1,123Updated 6 months ago