World Models Frameworks

Frameworks, implementations, and benchmarks for building and training world models that learn latent representations of environments for planning and imagination. Does NOT include general video generation, embodied AI applications without world model focus, or individual research papers without accompanying code/framework.

There are 23 world models frameworks tracked. 7 score above 50 (established tier). The highest-rated is galilai-group/stable-worldmodel at 62/100 with 200 stars. 2 of the top 10 are actively maintained.

Get all 23 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=world-models-frameworks&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	galilai-group/stable-worldmodel Reliable, minimal and scalable library for evaluating and conducting world...	62	Established	200	Python
2	NM512/dreamerv3-torch Implementation of Dreamer v3 in pytorch.	61	Established	813	Python
3	danijar/dreamerv2 Mastering Atari with Discrete World Models	60	Established	1,012	Python
4	danijar/dreamerv3 Mastering Diverse Domains through World Models	60	Established	2,917	Python
5	leofan90/Awesome-World-Models A comprehensive list of papers for the definition of World Models and using...	59	Established	1,334	—
6	danijar/dreamer Dream to Control: Learning Behaviors by Latent Imagination	59	Established	584	Python
7	LMD0311/Awesome-World-Model Collect some World Models for Autonomous Driving (and Robotic, etc.) papers.	50	Established	1,889	—
8	kc-ml2/SimpleDreamer A Simplified Pytorch Version of the Dreamer Algorithm	45	Emerging	151	Python
9	OpenRaiser/awesome-world-model-evolution A curated collection of research papers, models, and resources tracing the...	43	Emerging	148	—
10	duc0/deep-dream-in-pytorch Pytorch implementation of the DeepDream computer vision algorithm	40	Emerging	103	Jupyter Notebook
11	HaoranZhuExplorer/World-Models-Autonomous-Driving-Survey A curated list of world models for autonomous driving.	36	Emerging	498	—
12	AbdelStark/jepa-rs Joint Embedding Predictive Architecture for World Models, written in Rust.	35	Emerging	8	Rust
13	zhanghm1995/awesome-world-models-for-AD A curated list of awesome world models for autonomous driving (continually updated)	32	Emerging	12	—
14	anh-nn01/Time-Aware-World-Model Official Code for Time-Aware World Model (TAWM) with TD-MPC2 as the baseline...	27	Experimental	6	Python
15	CJReinforce/JOWA Official code for the ICLR 2025 paper, "Scaling Offline Model-Based RL via...	27	Experimental	28	Python
16	kyegomez/awesome-robotic-foundation-models A vast array of Multi-Modal Embodied Robotic Foundation Models!	27	Experimental	28	—
17	MatheusPereiraSilva/dream-framework A research-oriented Python framework implementing the DREAM architecture for...	22	Experimental	4	Python
18	OpenRaiser/CoW-Bench Leaderboard for The Trinity of Consistency as a Defining Principle for...	21	Experimental	9	HTML
19	Slownite/Trident-WM A high-performance World Model for robotics using PyTorch Lightning. It...	21	Experimental	—	Python
20	canbalkaya/Machine-Dreams-WWDC21 Machine Dreams is a playground exhibition that explores how a machine can...	19	Experimental	4	—
21	H-IAAC/WorldMachine Investigating the concept and creation of computational world models	17	Experimental	—	Jupyter Notebook
22	jeree02847/Dream Dream is a collaborative platform that helps users explore and share their...	14	Experimental	—	—
23	MukundaKatta/dreamnet DreamNet — AI Hallucination as Dreaming. Research platform studying LLM...	14	Experimental	—	Python

Comparisons in this category

dreamerv2 and dreamerv3 (60 vs 60) Awesome-World-Models and Awesome-World-Model (59 vs 50) Awesome-World-Model and World-Models-Autonomous-Driving-Survey (50 vs 36) Awesome-World-Model and awesome-world-model-evolution (50 vs 43) Awesome-World-Model and awesome-world-models-for-AD (50 vs 32) Awesome-World-Models and World-Models-Autonomous-Driving-Survey (59 vs 36) dreamerv2 and dreamer (60 vs 59) Awesome-World-Models and awesome-world-model-evolution (59 vs 43) Awesome-World-Models and awesome-world-models-for-AD (59 vs 32) dreamerv3-torch and SimpleDreamer (61 vs 45)