Reinforcement learning environments for compiler and program optimization tasks
☆994Feb 21, 2026Updated last week
Alternatives and similar repositories for CompilerGym
Users that are interested in CompilerGym are comparing it to the libraries listed below
Sorting:
- Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.☆761Feb 18, 2026Updated last week
- Must read research papers and links to tools and datasets that are related to using machine learning for compilers and systems optimisati…☆1,655Jan 21, 2026Updated last month
- A Graph-based Program Representation for Data Flow Analysis and Compiler Optimizations☆320May 22, 2024Updated last year
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,754Updated this week
- C/C++ frontend for MLIR. Also features polyhedral optimizations, parallel optimizations, and more!☆605Jun 19, 2025Updated 8 months ago
- ☆422Jan 4, 2026Updated last month
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,614Updated this week
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,733Oct 19, 2024Updated last year
- The Tensor Algebra SuperOptimizer for Deep Learning☆739Jan 26, 2023Updated 3 years ago
- ComPy-Learn is a framework for exploring program representations for ML4CODE tasks.☆22Aug 7, 2023Updated 2 years ago
- Development repository for the Triton language and compiler☆18,460Feb 22, 2026Updated last week
- Compiler for Neural Network hardware accelerators☆3,326May 11, 2024Updated last year
- High-performance automatic differentiation of LLVM and MLIR.☆1,542Feb 22, 2026Updated last week
- Research language for array processing in the Haskell/ML family☆1,664Jan 5, 2026Updated last month
- An extensible framework for program autotuning☆428Jan 29, 2026Updated last month
- Python interface for MLIR - the Multi-Level Intermediate Representation☆272Nov 28, 2024Updated last year
- Open Machine Learning Compiler Framework☆13,142Updated this week
- NeuroVectorizer is a framework that uses deep reinforcement learning (RL) to predict optimal vectorization compiler pragmas for for loops…☆99Jan 7, 2023Updated 3 years ago
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆976Feb 20, 2026Updated last week
- A Python compiler design toolkit.☆494Updated this week
- Implementation of IR2Vec, LLVM IR Based Scalable Program Embeddings☆101Feb 11, 2026Updated 2 weeks ago
- ☆95Nov 4, 2022Updated 3 years ago
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,762Apr 22, 2021Updated 4 years ago
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆1,006Sep 19, 2024Updated last year
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆182Apr 25, 2022Updated 3 years ago
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,349Apr 14, 2025Updated 10 months ago
- Bridging polyhedral analysis tools to the MLIR framework☆119Sep 9, 2023Updated 2 years ago
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- A collection of out-of-tree LLVM passes for teaching and learning☆3,359Jan 1, 2026Updated 2 months ago
- ☆250Jul 27, 2025Updated 7 months ago
- a language for fast, portable data-parallel computation☆6,577Updated this week
- Automatic verification of LLVM optimizations☆1,062Feb 12, 2026Updated 2 weeks ago
- 👨💻 My PhD.☆189Oct 5, 2022Updated 3 years ago
- Conversions to MLIR EmitC☆134Dec 12, 2024Updated last year
- Neural Code Comprehension: A Learnable Representation of Code Semantics☆216Nov 22, 2024Updated last year
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,751Dec 18, 2025Updated 2 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆125Jun 23, 2022Updated 3 years ago
- An MLIR frontend for tensor expressions☆24Sep 5, 2020Updated 5 years ago
- Circuit IR Compilers and Tools☆2,044Updated this week