DanHrmti / HECRL Star 16 Code Issues Pull requests Official PyTorch implementation of "Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion", Haramati et al., ICLR 2026 reinforcement-learning goal-conditioned-rl object-centric-learning hierarchical-rl diffusion-planning Updated Feb 23, 2026 Python
GradientSpaces / ECD Star 9 Code Issues Pull requests [ICML 2026] Energy-based Compositional Diffusion Planning robotics diffusers diffusion-policy diffusion-planning Updated Jun 23, 2026 Jupyter Notebook
mathisweil / minihack-ReMDM-planner Star 2 Code Issues Pull requests Masked diffusion (ReMDM) action planner for MiniHack in PyTorch. Code for "Return-Weighted ELBO Fine-Tuning Degrades Masked Diffusion Planners", NeurIPS 2026 BeNTo workshop. reinforcement-learning pytorch imitation-learning minihack discrete-diffusion masked-diffusion diffusion-planning neurips-2026 Updated Oct 6, 2026 Python
mathisweil / craftax-ReMDM-planner Star 1 Code Issues Pull requests Masked diffusion (ReMDM) action planner for Craftax Classic in JAX/Flax. Code for "Return-Weighted ELBO Fine-Tuning Degrades Masked Diffusion Planners", NeurIPS 2026 BeNTo workshop. reinforcement-learning flax imitation-learning jax discrete-diffusion masked-diffusion diffusion-planning craftax neurips-2026 Updated Oct 6, 2026 Python