95.47% on CIFAR10 with PyTorch

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

PyTorch implementation of adversarial attacks [torchattacks]

Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.

[NeurIPS'19] Deep Equilibrium Models

Compare neural networks by their feature similarity

Code for the MedRAG toolkit

A Type-1 Diabetes simulator implemented in Python for Reinforcement Learning purpose

A small library of LLM judges

Python ESPIRiT implementation.

Active Acquisition for fastMRI

Generate numerical phantoms.

Official implementation of ICML'24 paper "Offline Multi-Objective Optimization".

Source code for Generative Adversarial Bayesian Optimization (GABO) for Surrogate Objectives