Wavelets with JAX
Introduction
CR-Wavelets is a port of PyWavelets for Google JAX. It enables running Wavelet decomposition and reconstruction algorithms on GPU and TPU hardware.
For detailed documentation and usage, please visit online docs.
For theoretical background, please check online notes at Topics in Signal Processing and references therein (still under development).
CR-Wavelets is part of CR-Suite.
Related libraries:
Supported Platforms
CR-Wavelets can run on any platform supported by JAX. We have tested CR-Wavelets on Mac and Linux platforms and Google Colaboratory.
The latest code in the library has been tested against JAX 0.4.
JAX is not officially supported on Windows platforms at the moment. Although, it is possible to build it from source using Windows Subsystems for Linux. Alternatively, you can check out the community supported Windows build for JAX available from https://github.com/cloudhan/jax-windows-builder. This seems to work well and all the unit tests in the library have passed on Windows also.
Installation
Installation from PyPI:
python -m pip install cr-waveletsDirectly from our GITHUB repository:
python -m pip install git+https://github.com/carnotresearch/cr-wavelets.gitExamples/Usage
See the examples gallery in the documentation.
Contribution Guidelines/Code of Conduct
Documentation | Code | Issues | Discussions |
1 maintainer
Release files for cr-wavelets 0.4.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| cr-wavelets-0.4.0.tar.gz | 128.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| cr_wavelets-0.4.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 265.4 kB
Release files / cr-wavelets-0.4.0.tar.gz
| Download URL | cr-wavelets-0.4.0.tar.gz |
|---|---|
| Size | 128.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
51459ed358323143eccbf335b68f27f1c0b1c7a8387238b4689c2843466214a4
|
|
BLAKE2b-256 checksum How to use checksums |
bd5111eae7b4660afbe8338864aee4b51fba22c6b1f16eff61f8029cbbaa8fe6
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.8.18
|
Release files / cr_wavelets-0.4.0-py3-none-any.whl
| Download URL | cr_wavelets-0.4.0-py3-none-any.whl |
|---|---|
| Size | 137.4 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
1f6cd201d86fb27101535e0338a2bfc0676d8bc52ec672f617ee5ebd5f182736
|
|
BLAKE2b-256 checksum How to use checksums |
5248ddb3897042732e50d496705432b9923b0d1db51120af22a4573e240ffa1f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.8.18
|
