DIAMOND（DIffusion As a Model Of eNvironment Dreams）は、アータリゲームにおける視覚的な詳細が非常に重要なワールドモデリングのために、拡散ワールドモデルで訓練された強化学習エージェントです。自己回帰的想像を用いてアータリゲームのサブセットで訓練されており、事前訓練済みのワールドモデルを迅速にインストールして試すことができます。