DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model for visually-rich worlds crucial to Atari games. Trained on a subset of Atari games using autoregressive imagination, it offers quick installation and allows users to experiment with pre-trained world models.