Setup

To install MiniML, simply use pip or your package manager of choice:

pip install miniml-jax

You can use the cuda extras to install also a GPU-enabled version of Jax as a dependency:

pip install miniml-jax[cuda]

By default MiniML depends on a version of Jax that is not enabled for GPU or TPU. If you want to make use of hardware acceleration (see Jax's installation guide), you should update it separately with pip after having installed miniml-jax.