Nikola Milosevic's repositories
riemann-nn
Understanding the paper "Principles of Riemannian Geometry in Neural Networks" by Michael Hauser and Asok Ray
arc
ARC: A tool for creating artificial languages with rhythmicity control
CausalWorld
CausalWorld: A Robotic Manipulation Benchmark for Causal Structure and Transfer Learning
cleanrl
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
dsprites-dataset
Dataset to assess the disentanglement properties of unsupervised learning methods
jemris
JEMRIS is a versatile, open-source, multi-purpose MRI simulator.
mastering-urlb
[ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and use planning (Dyna-MPC) during fine-tuning.
milosen.github.io
Allan Lab website
pomdp-baselines
Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022
oderl
Experiment code for "Continuous-Time Model-Based Reinforcement Learning"
Safe-Policy-Optimization
NeurIPS 2023: Safe Policy Optimization: A benchmark repository for safe reinforcement learning algorithms