jeanmarcalkazzi / jaxmapp

JAX-based implementation for multi-agent path planning (MAPP) in continuous spaces.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

JAXMAPP: JAX-based Library for Multi-Agent Path Planning in Continuous Spaces

gallery roadmap metrics

JAXMAPP is a JAX-based library for multi-agent path planning (MAPP) in continuous spaces, with a particular emphasis on roadmap construction and evaluation. With JAXMAPP, You can:

  • Create MAPP problem instances with homogeneous/heterogeneous agents
  • Construct roadmaps and perform prioritized planning to solve MAPP
  • Develop and evaluate your own roadmap construction methods

Main contributors: Ryo Yonetani (@yonetaniryo), Keisuke Okumura (@Kei18)

Installation

The code has been tested on Ubuntu 16.04 and 18.04, as well as WSL2 (Ubuntu 20.04) on Windows 11. Planning can be performed only on the CPU, and the use of GPUs is also supported for training/evaluating machine-learning models. We also provide Dockerfile to replicate our setup.

venv

$ python -m venv .venv
$ source .venv/bin/activate
(.venv) $ make

Docker container

$ docker-compose build
$ docker-compose up -d dev
$ docker-compose exec dev bash

Docker container with CUDA enabled

$ docker-compose up -d dev-gpu
$ docker-compose exec dev-gpu bash

and update JAX modules in the container...

# pip install "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html

Tutorials

Citation

@misc{jaxmapp_2022,
author = {Yonetani, Ryo and Okumura, Keisuke},
month = {2},
title = {JAXMAPP: JAX-based Library for Multi-Agent Path Planning in Continuous Spaces},
url = {https://github.com/omron-sinicx/jaxmapp},
year = {2022}
}


@inproceedings{okumura2022ctrm,
title={CTRMs: Learning to Construct Cooperative Timed Roadmaps for Multi-agent Path Planning in Continuous Spaces},
author={Okumura, Keisuke and Yonetani, Ryo and Nishimura, Mai and Kanezaki, Asako},
booktitle={Proceedings of the International Conference on Autonomous Agents and Multiagent Systems (AAMAS)},
year={2022}
}

About

JAX-based implementation for multi-agent path planning (MAPP) in continuous spaces.

License:MIT License


Languages

Language:Jupyter Notebook 93.7%Language:Python 6.0%Language:Shell 0.1%Language:Dockerfile 0.1%Language:Cython 0.1%Language:Makefile 0.0%