Pre-quaxed libraries for multiple dispatch over abstract array types in JAX
Project description
quaxed
Pre-Quaxify
'ed JAX
Quaxed
wraps jax libraries (using
quax
) to enable using those libraries with
custom array-ish objects, not only jax arrays.
Installation
pip install quaxed
Documentation
Quick Start
To understand how quax
works it's magic, see
quax.quaxify
and the
tutorials.
To use this library, it's as simple as:
# Import pre-quaxified library
>>> import quaxed.numpy as jnp # this is quaxify(jax.numpy)
# As an example, let's import an array-ish object
>>> from unxt import Quantity
>>> x = Quantity(2, "km")
>>> jnp.square(w)
Quantity['area'](Array(4, dtype=int64, weak_type=True), unit='km2')
Development
We welcome contributions!
Citation
If you found this library to be useful and want to support the development and maintenance of lower-level utility libraries for the scientific community, please consider citing this work.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file quaxed-0.6.7.tar.gz
.
File metadata
- Download URL: quaxed-0.6.7.tar.gz
- Upload date:
- Size: 133.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 196f3d3a4d6d056c58cce3d78408c84ec1023a219489b5dc9d2920bddc3e9af7 |
|
MD5 | b7f3aa9655278c7a3b58d1bb21d4e0b1 |
|
BLAKE2b-256 | d3ecf943d565bf253007b8dd1e03f46e492fdde8540d2a00705257e48e9bfa71 |
Provenance
The following attestation bundles were made for quaxed-0.6.7.tar.gz
:
Publisher:
cd.yml
on GalacticDynamics/quaxed
-
Statement type:
https://in-toto.io/Statement/v1
- Predicate type:
https://docs.pypi.org/attestations/publish/v1
- Subject name:
quaxed-0.6.7.tar.gz
- Subject digest:
196f3d3a4d6d056c58cce3d78408c84ec1023a219489b5dc9d2920bddc3e9af7
- Sigstore transparency entry: 151056534
- Sigstore integration time:
- Predicate type:
File details
Details for the file quaxed-0.6.7-py3-none-any.whl
.
File metadata
- Download URL: quaxed-0.6.7-py3-none-any.whl
- Upload date:
- Size: 36.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 09fdb5446c4894bdc2b437d3b0458c0f8b746d366ac2e18ce76f6f1dd91efa15 |
|
MD5 | ad1ab6fd4d096720670cafb2707fe288 |
|
BLAKE2b-256 | f3881d1668b6545b83a408d48cf20545e173f2c3aee97f9bf88a08651440516b |
Provenance
The following attestation bundles were made for quaxed-0.6.7-py3-none-any.whl
:
Publisher:
cd.yml
on GalacticDynamics/quaxed
-
Statement type:
https://in-toto.io/Statement/v1
- Predicate type:
https://docs.pypi.org/attestations/publish/v1
- Subject name:
quaxed-0.6.7-py3-none-any.whl
- Subject digest:
09fdb5446c4894bdc2b437d3b0458c0f8b746d366ac2e18ce76f6f1dd91efa15
- Sigstore transparency entry: 151056535
- Sigstore integration time:
- Predicate type: