ondrejbiza

action_priors

Source code for Action Priors for Large Action Spaces in Robotics. Biza, Wang, Platt, van de Meent, Wong. AAMAS'21.

action_attention_iclr_22

Source code for "Binding Actions to Objects in World Models. Ondrej Biza, Robert Platt, Jan-Willem van de Meent, Lawson L. S. Wong and Thomas Kipf. ICLR 2022 workshop on Objects, Structure and Causality."

aamas_19

Source code for the paper "Online Abstraction with MDP Homomorphisms for Deep Learning".

negative-sampling-icml-21

Original source code for "The Impact of Negative Sampling on Contrastive Structured World Models. O Biza, E van der Pol, T Kipf. ICML'21 SSL workshop."

discrete_abstractions

Source code for Learning Discrete State Abstractions With Deep Variational Inference. Biza, Platt, van de Meent, Wong. AABI'21. https://arxiv.org/abs/2003.04300

bisim_metric

c-swm

Contrastive Learning of Structured World Models

relational_ndf

vae

Variational autoencoders implemented in Tensorflow.

baby-a3c

A high-performance Atari A3C agent in 180 lines of PyTorch

bmg

Bayesian mixture of Gaussians

cem

feudal-rl

fewshot

geom_cloning_corl_23

google-research

Google Research

helping_hands_website

hmm

Hidden Markov Models.

invariant_slot_attention.github.io

ondrejbiza.github.io

My personal website.

pgm

Exercises from the Probabilistic Graphical Models: Principles and Techniques book.

pytorch-maml-rl

Reinforcement Learning with Model-Agnostic Meta-Learning in Pytorch

se3-transformer-public

code for the SE3 Transformers paper: https://arxiv.org/abs/2006.10503

shapewarping

so3_examples

sonnet

TensorFlow-based neural network library

SPACE

Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"

stable-baselines

A fork of OpenAI Baselines, implementations of reinforcement learning algorithms

sunds

taxpose

TAX-Pose: Task-Specific Cross-Pose Estimation for Robot Manipulation