Skip to main content

Small collection of useful utilities for JAX

Project description

Powerpax: Utilities for JAX

Powerpax on PyPI Powerpax on conda-forge Documentation Tests

A collection of utility functions for JAX.

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.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

powerpax-0.1.2.tar.gz (10.0 kB view hashes)

Uploaded Source

Built Distribution

powerpax-0.1.2-py3-none-any.whl (9.5 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page