Smaller World Models for Reinforcement Learning Model-based reinforcement learning implemented in PyTorch. Trajectories generated by our world model:
