Skip to main content

Pre-quaxed libraries for multiple dispatch over abstract array types in JAX

Project description

quaxed

Pre-Quaxify'ed JAX

Quaxed wraps jax libraries (using quax) to enable using those libraries with custom array-ish objects, not only jax arrays.

Installation

PyPI version PyPI platforms

pip install quaxed

Documentation

Documentation Status

Quick Start

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 jnp  # this is quaxify(jax.numpy)

# As an example, let's import an array-ish object
>>> from unxt import Quantity
>>> x = Quantity(2, "km")
>>> jnp.square(w)
Quantity['area'](Array(4, dtype=int64, weak_type=True), unit='km2')

Development

Actions Status Codecov

We welcome contributions!

Citation

DOI

If you found this library to be useful and want to support the development and maintenance of lower-level utility libraries for the scientific community, please consider citing this work.

Project details


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.6.7.tar.gz (133.5 kB view details)

Uploaded Source

Built Distribution

quaxed-0.6.7-py3-none-any.whl (36.1 kB view details)

Uploaded Python 3

File details

Details for the file quaxed-0.6.7.tar.gz.

File metadata

  • Download URL: quaxed-0.6.7.tar.gz
  • Upload date:
  • Size: 133.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/5.1.1 CPython/3.12.7

File hashes

Hashes for quaxed-0.6.7.tar.gz
Algorithm Hash digest
SHA256 196f3d3a4d6d056c58cce3d78408c84ec1023a219489b5dc9d2920bddc3e9af7
MD5 b7f3aa9655278c7a3b58d1bb21d4e0b1
BLAKE2b-256 d3ecf943d565bf253007b8dd1e03f46e492fdde8540d2a00705257e48e9bfa71

See more details on using hashes here.

Provenance

The following attestation bundles were made for quaxed-0.6.7.tar.gz:

Publisher: cd.yml on GalacticDynamics/quaxed

Attestations:

File details

Details for the file quaxed-0.6.7-py3-none-any.whl.

File metadata

  • Download URL: quaxed-0.6.7-py3-none-any.whl
  • Upload date:
  • Size: 36.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/5.1.1 CPython/3.12.7

File hashes

Hashes for quaxed-0.6.7-py3-none-any.whl
Algorithm Hash digest
SHA256 09fdb5446c4894bdc2b437d3b0458c0f8b746d366ac2e18ce76f6f1dd91efa15
MD5 ad1ab6fd4d096720670cafb2707fe288
BLAKE2b-256 f3881d1668b6545b83a408d48cf20545e173f2c3aee97f9bf88a08651440516b

See more details on using hashes here.

Provenance

The following attestation bundles were made for quaxed-0.6.7-py3-none-any.whl:

Publisher: cd.yml on GalacticDynamics/quaxed

Attestations:

Supported by

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