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.8.tar.gz
(43.6 kB
view details)
Built Distribution
quaxed-0.4.8-py3-none-any.whl
(30.7 kB
view details)
File details
Details for the file quaxed-0.4.8.tar.gz.
File metadata
- Download URL: quaxed-0.4.8.tar.gz
- Upload date:
- Size: 43.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | bda4257b223513e53681623b434e52ef6f36c8b5179519b359643c1d05a5cb6c |
|
| MD5 | 47341b93af902f6bf9403c33b9cca7f2 |
|
| BLAKE2b-256 | 193d79c5c508cf650afa40c4b77f34be5c0283d8ad6ce5669756f281d66c060e |
File details
Details for the file quaxed-0.4.8-py3-none-any.whl.
File metadata
- Download URL: quaxed-0.4.8-py3-none-any.whl
- Upload date:
- Size: 30.7 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 | 059e825e2e3f9896812e3f3f922ef976754633609a7324d8a709f1efe4d2c5ce |
|
| MD5 | 054a50134a4dc6e6ab33d1c6606d0b1d |
|
| BLAKE2b-256 | 43293dcc36f280d3f7a8099824b228ee864e4b485cbdb713ff28ee96ef0a59be |
