├── .gitignore ├── LICENSE ├── README.md ├── assets └── joint.gif ├── mbd ├── __init__.py ├── assets │ ├── car2d_xref.npy │ ├── cartpole.xml │ ├── humanoidrun.xml │ ├── humanoidstandup.xml │ ├── humanoidtrack.xml │ ├── jog_xref.pkl │ ├── pushT.xml │ └── walk_xref.pkl ├── blackbox │ ├── mbd_mnist.py │ └── mbd_opt.py ├── envs │ ├── __init__.py │ ├── car2d.py │ ├── cartpole.py │ ├── hopper.py │ ├── humanoidrun.py │ ├── humanoidstandup.py │ ├── humanoidtrack.py │ ├── pushT.py │ └── walker2d.py ├── notebooks │ └── 01_1d_demo.py ├── planners │ ├── __init__.py │ ├── mbd_planner.py │ └── path_integral.py ├── rl │ └── train_brax.py ├── scripts │ ├── run_mbd.py │ ├── vis_diffusion.py │ └── vis_manim.py └── utils.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /assets/joint.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/assets/joint.gif -------------------------------------------------------------------------------- /mbd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/__init__.py -------------------------------------------------------------------------------- /mbd/assets/car2d_xref.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/car2d_xref.npy -------------------------------------------------------------------------------- /mbd/assets/cartpole.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/cartpole.xml -------------------------------------------------------------------------------- /mbd/assets/humanoidrun.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/humanoidrun.xml -------------------------------------------------------------------------------- /mbd/assets/humanoidstandup.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/humanoidstandup.xml -------------------------------------------------------------------------------- /mbd/assets/humanoidtrack.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/humanoidtrack.xml -------------------------------------------------------------------------------- /mbd/assets/jog_xref.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/jog_xref.pkl -------------------------------------------------------------------------------- /mbd/assets/pushT.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/pushT.xml -------------------------------------------------------------------------------- /mbd/assets/walk_xref.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/assets/walk_xref.pkl -------------------------------------------------------------------------------- /mbd/blackbox/mbd_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/blackbox/mbd_mnist.py -------------------------------------------------------------------------------- /mbd/blackbox/mbd_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/blackbox/mbd_opt.py -------------------------------------------------------------------------------- /mbd/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/__init__.py -------------------------------------------------------------------------------- /mbd/envs/car2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/car2d.py -------------------------------------------------------------------------------- /mbd/envs/cartpole.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/cartpole.py -------------------------------------------------------------------------------- /mbd/envs/hopper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/hopper.py -------------------------------------------------------------------------------- /mbd/envs/humanoidrun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/humanoidrun.py -------------------------------------------------------------------------------- /mbd/envs/humanoidstandup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/humanoidstandup.py -------------------------------------------------------------------------------- /mbd/envs/humanoidtrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/humanoidtrack.py -------------------------------------------------------------------------------- /mbd/envs/pushT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/pushT.py -------------------------------------------------------------------------------- /mbd/envs/walker2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/envs/walker2d.py -------------------------------------------------------------------------------- /mbd/notebooks/01_1d_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/notebooks/01_1d_demo.py -------------------------------------------------------------------------------- /mbd/planners/__init__.py: -------------------------------------------------------------------------------- 1 | from . import mbd_planner, path_integral -------------------------------------------------------------------------------- /mbd/planners/mbd_planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/planners/mbd_planner.py -------------------------------------------------------------------------------- /mbd/planners/path_integral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/planners/path_integral.py -------------------------------------------------------------------------------- /mbd/rl/train_brax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/rl/train_brax.py -------------------------------------------------------------------------------- /mbd/scripts/run_mbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/scripts/run_mbd.py -------------------------------------------------------------------------------- /mbd/scripts/vis_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/scripts/vis_diffusion.py -------------------------------------------------------------------------------- /mbd/scripts/vis_manim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/scripts/vis_manim.py -------------------------------------------------------------------------------- /mbd/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/mbd/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeCAR-Lab/model-based-diffusion/HEAD/setup.py --------------------------------------------------------------------------------