# Installation

First, make sure that you have a working installation of JAX (as explained [here](https://jax.readthedocs.io/en/latest/developer.html#installing-jax))

Then, use the following command will install `augmax` from PyPI:

```bash
pip install augmax
```

In case you want to use the latest version from github directly, please run

```bash
pip install git+https://github.com/khdlr/augmax
```