jonasrothfuss

python-sortedcontainers

Python Sorted Container Types: Sorted List, Sorted Dict, and Sorted Set

ml_logger

A logger, server and visualization dashboard for ML projects

video2tfrecord

Easily convert RGB video data (e.g. .avi) to the TensorFlow tfrecords file format for training e.g. a NN in TensorFlow. This implementation allows to limit the number of frames per video to be stored in the tfrecords.

tensorflow-maml

TensorFlow 2.0 implementation of MAML.

simulator

Spatiotemporal epidemic model introduced in the context of COVID-19, ACM TSAS, 2022

doodad

A job launching library for docker, EC2, GCP, etc.

blackbox_mpc

Sampling based Model Predictive Control package for Model-Based RL research

avici

Amortized Inference for Causal Structure Learning, NeurIPS 2022

pypmc

Clustering with variational Bayes and population Monte Carlo

KernelMixtureNetwork

Kernel Mixture Network implementation with some minor tweaks

dibs

DiBS: Differentiable Bayesian Structure Learning, NeurIPS 2021

movement_primitives_via_optimization

Implementation of the paper "Movement Primitives via Optimization" (Dragan et al., 2016). It includes both the adaptation of trajectories with DMP and learning a better adaptation norm.

rllib

learningdynamics

The code for the NeurIPS 2019 Graph Representation Learning workshop paper "Learning Visual Dynamics Models of Rigid Objects using Relational Inductive Biases" (Ferreira et al., 2019)

adaptive-constraint-learning

Code accompanying the paper "Interactively Learning Preference Constraints in Linear Bandits" (ICML 2022).

parameterized_model

TensorFlow parameterized model library

Model-based-policy-optimizers

Model based policy optimizers

pacoh-jax

Implementation of PACOH and F-PACOH from Rothfuss et al. (2021)