Machine Learning and Mind Lab (MLML)

Popular repositories Loading

1. mctd mctd Public

   Official Code for "Monte Carlo Tree Diffusion for System 2 Planning" and "Fast Monte Carlo Tree Diffusion: 100x Speedup via Parallel and Sparse Planning"

   Python 30 8

2. dreamweaver-release dreamweaver-release Public

   Official Code for "Dreamweaver: Learning Compositional World Models from Pixels"

   Python 11 2

3. lddm lddm Public

   [ICLR2026] Official Code for Loopholing Discrete Diffusion: Deterministic Bypass of the Sampling Wall

   Python 10 1

4. gram-release gram-release Public

   Official codebase for the paper: "Generative Recursive Reasoning"

   Python 9 2

5. drstrategy drstrategy Public

   This is the official code for the paper "Dr. Strategy: Model-Based Generalist Agents with Strategic Dreaming" (ICML 2024).

   Python 8

6. learning-to-theorize learning-to-theorize Public

   Python 8 1