World Models Python Packages

dreamer4

Implementation of Danijar's latest iteration for his Dreamer line of work

x-transformers-rl

Implementation of a transformer for reinforcement learning using `x-transformers`

improving-transformers-world-model

Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch

gaia2-pytorch

Gaia2 - Pytorch

genie2-pytorch

Implementation of a framework for Genie2 in Pytorch

grilly

GPU-accelerated neural network operations using Vulkan compute shaders

gemla

Γ–EML–α transport architecture SDK for lifted-phase diagnostics, action-based inference, and adversarial-control gates.

dreamerv3

Mastering Diverse Domains through World Models

dreamerv2

Mastering Atari with Discrete World Models

worldforge-ai

Testable world-model workflows for physical-AI systems.

ewm-core

Observable trajectory infrastructure — trading domain reference implementation

dreamer

Mastering Diverse Domains through World Models

loci-stdb

AI memory for the physical world — 4D spatiotemporal vector database that remembers where things were, not just what's visible now. Hilbert bucketing · temporal sharding · predict-then-retrieve · novelty detection.

mmlm

Toward Multi Modality Language Model - implementation of GPT-4o/Project Astra

pixyzrl

A Bayesian RL Framework with Probabilistic Generative Models

worldflux

Unified Python interface for world models in reinforcement learning