Chargax: A JAX Accelerated EV Charging Simulator

---

📦 Installation

For those using uv, it is possible to run a standard PPO implementation with default settings by directly running uv run main.py.

git clone git@github.com:ponseko/chargax.git
cd chargax
uv run main.py

Alternatively, install the project as an editable package in your favourite virtual environment software. E.g. using conda:

git clone git@github.com:ponseko/chargax.git
cd chargax
conda create -n chargax python=3.11
conda activate chargax
pip install -e .

python main.py

for CUDA support, additionally run pip install jax[cuda].

---

📑 Citing

@misc{ponse2025chargaxjaxacceleratedev,
      title={Chargax: A JAX Accelerated EV Charging Simulator}, 
      author={Koen Ponse, Jan Felix Kleuker, Aske Plaat, Thomas Moerland},
      year={2025},
      eprint={2507.01522},
      archivePrefix={arXiv},
      primaryClass={cs.LG},
      url={https://arxiv.org/abs/2507.01522}, 
}