├── .gitignore ├── README.md ├── jax_mppi ├── __init__.py ├── controllers │ ├── __init__.py │ ├── base.py │ ├── cemppi.py │ └── mppi.py ├── environments │ ├── __init__.py │ └── pendulum.py ├── rollout.py └── utils.py ├── main.py ├── poetry.lock ├── pyproject.toml └── video └── pendulum.gif /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/README.md -------------------------------------------------------------------------------- /jax_mppi/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.1.0' 2 | -------------------------------------------------------------------------------- /jax_mppi/controllers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/jax_mppi/controllers/__init__.py -------------------------------------------------------------------------------- /jax_mppi/controllers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/jax_mppi/controllers/base.py -------------------------------------------------------------------------------- /jax_mppi/controllers/cemppi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/jax_mppi/controllers/cemppi.py -------------------------------------------------------------------------------- /jax_mppi/controllers/mppi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/jax_mppi/controllers/mppi.py -------------------------------------------------------------------------------- /jax_mppi/environments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /jax_mppi/environments/pendulum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/jax_mppi/environments/pendulum.py -------------------------------------------------------------------------------- /jax_mppi/rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/jax_mppi/rollout.py -------------------------------------------------------------------------------- /jax_mppi/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/jax_mppi/utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/main.py -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/pyproject.toml -------------------------------------------------------------------------------- /video/pendulum.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jlehtomaa/jax-mppi/HEAD/video/pendulum.gif --------------------------------------------------------------------------------