Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions"

DSPy: The framework for programming—not prompting—language models

Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"

Official code for "QueST: Self-Supervised Skill Abstractions for Continuous Control" [NeurIPS 2024]

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

🌎💪 BrowserGym, a Gym environment for web task automation

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…

🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

PyTorch implementation of soft actor critic

Dobb·E: An open-source, general framework for learning household robotic manipulation

Friends don't let friends make certain types of data visualization - What are they and why are they bad.

The term project, SNU Robot Mechanics, FALL 2019

Clean implementation of conditional and unconditional behavior transformer.

Code and website for Behavior Transformers: Cloning k modes with one stone.

A curated list of Diffusion Model in RL resources (continually updated)

[ICLR 2023] Choreographer: a model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able to efficiently coordinate and adapt the skills to solve downstream …

2023 Spring SNU Computer Vision Project

Code for "Demonstration-free Autonomous Reinforcement Learning via Implicit and Bidirectional Curriculum" (ICML 2023)

Simple maze environments using mujoco-py

Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)

Acceptance rates for the major AI conferences