2 repositories on SrcLog
DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 Spotlight.
Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.