2025 Python Packaging Survey is now live!  Take the survey now

Array-API JAX compatibility

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for adrn from gravatar.com adrn Avatar for nstarman from gravatar.com nstarman

Unverified details

These details have not been verified by PyPI
Project links
Meta
  • License: BSD License
  • Author: Nathaniel Starkman
  • Requires: Python >=3.10
  • Provides-Extra: dev , docs , test
Classifiers
Report project as malware

Project description

quaxed

Actions Status Documentation Status

PyPI version Conda-Forge PyPI platforms

DOI

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')

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page