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.2.tar.gz
(42.9 kB
view details)
Built Distribution
quaxed-0.4.2-py3-none-any.whl
(30.0 kB
view details)
File details
Details for the file quaxed-0.4.2.tar.gz.
File metadata
- Download URL: quaxed-0.4.2.tar.gz
- Upload date:
- Size: 42.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.0.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 3300aab3840b071d1ea69d138d2be8c2e54708c322d6719224d0ce7ca366a587 |
|
| MD5 | 6cccb2989167213d17112377f5562cfd |
|
| BLAKE2b-256 | cde36b01039344f06a92ef6fe3d5355048df50a25af216fc2386b1bdb726b48d |
File details
Details for the file quaxed-0.4.2-py3-none-any.whl.
File metadata
- Download URL: quaxed-0.4.2-py3-none-any.whl
- Upload date:
- Size: 30.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.0.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 0fdd71e7bc71f8b3a725348ece3219e2179b94b2d5e232391e458e0328b2c73e |
|
| MD5 | a242e7670258e6b9b6f11ccffce2c8ac |
|
| BLAKE2b-256 | 68d3d94015fc09f414453fcacf945a5334d0e36fdd31dd1ac02cadc2fdf986a8 |
