DIAMOND (DIffusion As a Model Of eNvironment Dreams) est un agent d'apprentissage par renforcement entraîné sur un modèle de diffusion du monde, conçu pour la modélisation du monde dans les jeux Atari où les détails visuels sont cruciaux. Il est entraîné sur un sous-ensemble de jeux Atari à l'aide d'une imagination autorégressive et permet une installation rapide et l'expérimentation de modèles de monde pré-entraînés.