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 25 world models frameworks tracked. 5 score above 50 (established tier). The highest-rated is danijar/dreamerv3 at 66/100 with 2,917 stars and 412 monthly downloads. 3 of the top 10 are actively maintained.

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

Comparisons in this category

dreamerv3 and dreamerv2 (66 vs 58) Awesome-World-Models and Awesome-World-Model (49 vs 46) Awesome-World-Model and World-Models-Autonomous-Driving-Survey (46 vs 29) Awesome-World-Model and awesome-world-models-for-AD (46 vs 25) dreamerv2 and dreamer (58 vs 56) Awesome-World-Models and awesome-world-model-evolution (49 vs 39) Awesome-World-Models and awesome-world-models-for-AD (49 vs 25) dreamerv3-torch and SimpleDreamer (60 vs 38)