Skip to content

@ahn-ml Machine Learning and Mind Lab (MLML)

Change the repository type filter

All

    Repositories list

    12 repositories

    • vqgan

      Public
      Taming Transformers for High-Resolution Image Synthesis
      Jupyter Notebook
      •
      MIT License
      •1.2k•0•0•0•Updated Sep 22, 2025Sep 22, 2025

    • 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
      •
      Other
      •4•19•0•0•Updated Aug 21, 2025Aug 21, 2025

    • ctm

      Public
      Continuous Thought Machines, because thought takes time and reasoning is a process.
      Python
      •
      Apache License 2.0
      •272•0•0•0•Updated Jun 30, 2025Jun 30, 2025

    • dreamweaver-release

      Public
      Official Code for "Dreamweaver: Learning Compositional World Models from Pixels"
      Python
      •
      MIT License
      •0•8•0•0•Updated Jun 12, 2025Jun 12, 2025

    • drstrategy

      Public
      This is the official code for the paper "Dr. Strategy: Model-Based Generalist Agents with Strategic Dreaming" (ICML 2024).
      reinforcement-learningunsupervised-reinforcement-learning
      Python
      •0•9•0•0•Updated Nov 24, 2024Nov 24, 2024

    • openreview-crawl-citation-stats

      Public
      Python
      •0•0•0•0•Updated Nov 22, 2024Nov 22, 2024

    • gpu_testing

      Public
      To find the failing GPUs in a server.
      Python
      •0•2•0•0•Updated Dec 13, 2022Dec 13, 2022

    • crafter

      Public
      Benchmarking the Spectrum of Agent Capabilities
      Python
      •
      MIT License
      •87•0•0•0•Updated Mar 23, 2022Mar 23, 2022

    • OCVT

      Public
      Official PyTorch implementation of "Generative Video Transformer: Can Objects be the Words?"
      0•2•1•0•Updated Jun 11, 2021Jun 11, 2021

    • SCALOR

      Public
      Official PyTorch implementation of "SCALOR: Generative World Models with Scalable Object Representations"
      Python
      •
      MIT License
      •14•0•0•0•Updated Oct 2, 2020Oct 2, 2020

    • G-SWM

      Public
      Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"
      Python
      •
      GNU General Public License v3.0
      •6•0•0•0•Updated Aug 23, 2020Aug 23, 2020

    • SPACE

      Public
      Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"
      Python
      •
      GNU General Public License v3.0
      •29•1•0•0•Updated Apr 25, 2020Apr 25, 2020