← all repositories

eloialonso/iris

A reinforcement learning agent that learns world models via discrete autoencoders and transformers to achieve sample-efficient game-playing on Atari.

891 stars Python AgentsML Frameworks
iris
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

IRIS (Imaginary World Models for Efficient RL) is a model-based reinforcement learning agent that learns dynamics through two components: a discrete autoencoder that compresses observations into token representations, and an autoregressive transformer that predicts future token sequences. The agent trains entirely in imagination using the learned world model, enabling sample-efficient learning. Published at ICLR 2023.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.