SuperSonic, a new open-source framework to allow compiler developers to integrate RL into compilers easily, regardless of their RL expertise

Last update: Nov 28, 2022

Overview

Automating reinforcement learning architecture design for code optimization.

Check the website for more information.

Introduction

SUPERSONIC is a tool to automate reinforcement learning (RL) policy searching and tuning. It is designed to help compiler developers to find the right RL architecture and algorithm to use for an optimization task. It finds the right RL exploration algorithm, the reward function and a method for modeling the environment state.

Given an RL search space defined by the Supersonic Python API, the Supersonic meta-optimizer automatically searches for a suitable RL component composition for an optimization task. It also automatically tunes a set of tunable hyperparameters of the chosen components. No RL expertise is needed to use Supersonic.

Check our paper for detailed information.

Installation

Supersonic builds upon:

Python v3.8
CompilerGym
Ray

The system was tested on the following operating systems:

Ubuntu 18.04
Fedora 28
Debian 10

See INSTALL.md for further details.

Useage

See artifact_evaluation for an out-of-the-box demo of Supersonic.

Contributing

We welcome contributions to SuperSonic. If you are interested in contributing please see this document.

SuperSonic, a new open-source framework to allow compiler developers to integrate RL into compilers easily, regardless of their RL expertise

Related tags

Overview

Introduction

Installation

Useage

Contributing

Owner

Discerning Decision-Making Process of Deep Neural Networks with Hierarchical Voting Transformation

Offcial implementation of "A Hybrid Video Anomaly Detection Framework via Memory-Augmented Flow Reconstruction and Flow-Guided Frame Prediction, ICCV-2021".

Code for "Neural Parts: Learning Expressive 3D Shape Abstractions with Invertible Neural Networks", CVPR 2021

Official implementation of "Towards Good Practices for Efficiently Annotating Large-Scale Image Classification Datasets" (CVPR2021)

Custom implementation of Corrleation Module

Official implementation of "Accelerating Reinforcement Learning with Learned Skill Priors", Pertsch et al., CoRL 2020

Differentiable Abundance Matching With Python

A scikit-learn compatible neural network library that wraps PyTorch

improvement of CLIP features over the traditional resnet features on the visual question answering, image captioning, navigation and visual entailment tasks.

On Nonlinear Latent Transformations for GAN-based Image Editing - PyTorch implementation

PaRT: Parallel Learning for Robust and Transparent AI

CPF: Learning a Contact Potential Field to Model the Hand-object Interaction

A toolkit for developing and comparing reinforcement learning algorithms.

[PyTorch] Official implementation of CVPR2021 paper "PointDSC: Robust Point Cloud Registration using Deep Spatial Consistency". https://arxiv.org/abs/2103.05465

A tensorflow implementation of GCN-LPA

Pytorch Implementation of rpautrat/SuperPoint

RuDOLPH: One Hyper-Modal Transformer can be creative as DALL-E and smart as CLIP

Episodic-memory - Ego4D Episodic Memory Benchmark

Implementation of FSGNN

Bayesian Optimization using GPflow