Installation

prax is created based on jax, and please install jax at first. See JAX install guide for installation.

After the installation of jax, prax can be installed with pip directly from GitHub, with the following command:

pip install git+https://github.com/yonesuke/prax.git