The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Official inference repo for FLUX.1 models

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Hydra is a framework for elegantly configuring complex applications

Google Drive Public File Downloader when Curl/Wget Fails

Implementations of basic RL algorithms with minimal lines of codes! (pytorch based)

Reinforcement Learning in PyTorch

Gin provides a lightweight configuration framework for Python

Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT (CVPR'24), OA-CNNs (CVPR'24), MSC (CVPR'23)

Python 3.8+ toolbox for submitting jobs to Slurm

PyTorch implementation of Deep Reinforcement Learning: Policy Gradient methods (TRPO, PPO, A2C) and Generative Adversarial Imitation Learning (GAIL). Fast Fisher vector product TRPO.

MetaDrive: Open-source driving simulator

CLI and curses mixer for PulseAudio

Structural implementation of RL key algorithms

PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.

Use PyTorch Models with CasADi for data-driven optimization or learning-based optimal control. Supports Acados.

Create powerful Hydra applications without the yaml files and boilerplate code.

Reinforcement Learning with Model Predictive Control

Pyrallis is a framework for structured configuration parsing from both cmd and files. Simply define your desired configuration structure as a dataclass and let pyrallis do the rest!

Use PyTorch Models with CasADi and Acados

[CVPR'23] A Simple Framework for Text-Supervised Semantic Segmentation

Python-CasADi-based package containing several algorithms for solving mixed-integer nonlinear programs (MINLPs)

A dual-control effect preserving formulation for nonlinear output-feedback stochastic model predictive control with constraints

EPFL Verifier for Approximate Neural Networks and QPs

Modular Deep Reinforcement Learning libary based on PyTorch

make image files for the NYU Depth Dataset V2