Array-API JAX compatibility
Project description
quaxed
Quaxed wraps jax libraries (using
quax) to enable using those libraries with
custom array-ish objects, not only jax arrays.
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 qnp # this is quaxify(jax.numpy)
# As an example, let's import an array-ish object
>>> from unxt import Quantity
>>> x = Quantity(2, "km")
>>> qnp.square(w)
Quantity['area'](Array(4, dtype=int64, weak_type=True), unit='km2')
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
quaxed-0.4.5.tar.gz
(43.3 kB
view details)
Built Distribution
quaxed-0.4.5-py3-none-any.whl
(30.2 kB
view details)
File details
Details for the file quaxed-0.4.5.tar.gz.
File metadata
- Download URL: quaxed-0.4.5.tar.gz
- Upload date:
- Size: 43.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 9f1dd51eed33759d7fa2a32ab77958082d9ec6d3d2ecf5b12f63fbbedf8ce831 |
|
| MD5 | 597c4baf9126e578ba0c209b0d811410 |
|
| BLAKE2b-256 | 6b1621b80c411f0bc9190d223782e70bb4ef724bbf08b3c4520e9832c3524235 |
File details
Details for the file quaxed-0.4.5-py3-none-any.whl.
File metadata
- Download URL: quaxed-0.4.5-py3-none-any.whl
- Upload date:
- Size: 30.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 8aa8718660711b24f337db3cdbca8581a7630c2c39b25e2cce1a12a4d3e5ce91 |
|
| MD5 | ce52e0c4d1e74eacdf2d5e321f53c32f |
|
| BLAKE2b-256 | 01aca5ddbcd8998631dd7a8f3365916bb40bfb2249c87f74a2fc1680710b2a2d |
