← all repositories

jannerm/diffuser

A diffusion model library for training and planning via value-guided sampling in reinforcement learning environments.

1.3k stars Python ML Frameworks
diffuser
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

This repository implements diffusion models for flexible behavior synthesis in reinforcement learning settings. It provides code for training diffusion policies and planning through value-function guided sampling on D4RL locomotion tasks, block-stacking (KUKA), and Maze2D goal-reaching via inpainting. The project includes pretrained model downloads and integrates with the Hugging Face Diffusers ecosystem.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.