Powerpax: Utilities for JAX

A collection of utility functions for JAX.

• Documentation https://powerpax.readthedocs.io/en/latest/
• Source Code: https://github.com/karlotness/powerpax
• Bug Reports: https://github.com/karlotness/powerpax/issues

Installation

Install from PyPI using pip:

$ python -m pip install powerpax

or from conda-forge:

$ conda install -c conda-forge powerpax

This should install required dependencies, but JAX itself may require special attention, particularly for GPU support. Follow the JAX installation instructions.

License

This software is distributed under the MIT license. See LICENSE.txt for the license text.