Search 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

grilly

GPU-accelerated neural network operations using Vulkan compute shaders

genie2-pytorch

Genie2

dreamerv2

Mastering Atari with Discrete World Models

dreamerv3

Mastering Diverse Domains through World Models

worldforge-ai

Testable world-model workflows for physical-AI systems.

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.

pixyzrl

A Bayesian RL Framework with Probabilistic Generative Models

dreamer

Mastering Diverse Domains through World Models

mmlm

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

worldflux

Unified Python interface for world models in reinforcement learning