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')
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
quaxed-0.4.2.tar.gz
(42.9 kB
view hashes)
Built Distribution
quaxed-0.4.2-py3-none-any.whl
(30.0 kB
view hashes)