Last updated:
0 purchases
powerpax 0.1.2
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.
For personal and professional use. You cannot resell or redistribute these repositories in their original state.
There are no reviews.