AndyShih12

DreamPropeller

IF

Reflected-Diffusion

[ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)

paradigms

PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight

llama

Inference code for Llama models

LongHorizonTemperatureScaling

PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023

RL4LMs

A modular RL library to fine-tune language models to human preferences

vizier

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

mac

PyTorch implementation for "Training and Inference on Any-Order Autoregressive Models the Right Way", NeurIPS 2022 Oral, TPM 2023 Best Paper Honorable Mention

Conventions-ModularPolicy

PyTorch implementation for "On the Critical Role of Conventions in Adaptive Human-AI Collaboration", ICLR 2021

HyperSPN

PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021

PantheonRL

PantheonRL is a package for training and testing multi-agent reinforcement learning environments. PantheonRL supports cross-play, fine-tuning, ad-hoc coordination, and more.

probabilistic-circuits

A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.

gym-cooking

🏆 gym-cooking: Code for "Too many cooks: Bayesian inference for coordinating multi-agent collaboration", Winner of the CogSci 2020 Computational Modeling Prize in High Cognition, and a NeurIPS 2020 CoopAI Workshop Best Paper.

SPN_Variational_Inference

PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020

stable-baselines3

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

cheatsheets

Official Matplotlib cheat sheets

SSDC

Smoothing Structured Decomposable Circuits

SPFlow

Sum Product Flow: An Easy and Extensible Library for Sum-Product Networks