Metadata-Version: 2.1
Name: powerpax
Version: 0.1.2
Summary: Small collection of useful utilities for JAX
Author: Karl Otness
Requires-Python: >=3.9
Description-Content-Type: text/markdown
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Dist: jax>=0.4.6
Project-URL: Bug Tracker, https://github.com/karlotness/powerpax/issues
Project-URL: Documentation, https://powerpax.readthedocs.io
Project-URL: Homepage, https://github.com/karlotness/powerpax
Project-URL: Source Code, https://github.com/karlotness/powerpax

# Powerpax: Utilities for JAX

[![Powerpax on PyPI](https://img.shields.io/pypi/v/powerpax)][pypi]
[![Powerpax on conda-forge](https://img.shields.io/conda/vn/conda-forge/powerpax.svg)][condaforge]
[![Documentation](https://readthedocs.org/projects/powerpax/badge/?version=latest)][docs]
[![Tests](https://github.com/karlotness/powerpax/actions/workflows/test.yml/badge.svg)][tests]

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][pypi] using pip:
```console
$ python -m pip install powerpax
```
or from [conda-forge][condaforge]:
``` console
$ 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](https://jax.readthedocs.io/en/latest/installation.html).

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

[pypi]: https://pypi.org/project/powerpax/
[condaforge]: https://anaconda.org/conda-forge/powerpax
[docs]: https://powerpax.readthedocs.io/en/latest/
[tests]: https://github.com/karlotness/powerpax/actions/workflows/test.yml

