autoray is a lightweight python AUTOmatic-arRAY library for
abstracting your tensor operations. Primarily it provides an
automatic dispatch mechanism
that means you can write backend agnostic code that works for:
- numpy
- pytorch
- jax
- cupy
- mlx
- dask
- autograd
- tensorflow
- sparse
- paddle
- ... and indeed any library that provides a numpy-ish api, even if it
knows nothing about
autoray.
Beyond that, abstracting the array interface allows you to:
- swap custom versions of functions for specific backends
- trace through computations lazily without actually running them
- automatically share intermediates and fold constants in computations
- compile functions with a unified interface for different backends
Basic usage
The main function of autoray is
do,
which takes a function
name followed by *args and **kwargs, and automatically looks up (and
caches) the correct function to match the equivalent numpy call:
import autoray as ar
def noised_svd(x):
# automatic dispatch based on supplied array
U, s, VH = ar.do("linalg.svd", x)
# automatic dispatch based on different array
sn = s + 0.1 * ar.do("random.normal", size=ar.shape(s), like=s)
# automatic dispatch for multiple arrays for certain functions
return ar.do("einsum", "ij,j,jk->ik", U, sn, VH)
# explicit backend given by string
x = ar.do("random.uniform", size=(100, 100), like="torch")
# this function now works for any backend
y = noised_svd(x)
# explicit inference of backend from array
ar.infer_backend(y)
# 'torch'
If you don't like the explicit do syntax, or simply want a
drop-in replacement for existing code, you can also import the autoray.numpy
module:
from autoray import numpy as np
# set a temporary default backend
with ar.backend_like("cupy"):
z = np.ones((3, 4), dtype="float32")
np.exp(z)
# array([[2.7182817, 2.7182817, 2.7182817, 2.7182817],
# [2.7182817, 2.7182817, 2.7182817, 2.7182817],
# [2.7182817, 2.7182817, 2.7182817, 2.7182817]], dtype=float32)
Alternatively you can use
autoray.get_namespace
to get a backend specific (with optional default device and dtype) namespace
object, (c.f. the Python Array Api):
xp = ar.get_namespace(z)
xp.einsum("ii->i", z)
Custom backends and functions can be dynamically registered with:
The main documentation is available at autoray.readthedocs.io.
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file autoray-0.10.1.tar.gz.
File metadata
- Download URL: autoray-0.10.1.tar.gz
- Upload date:
- Size: 1.4 MB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
twine/6.1.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b158904f410757f326d825e57b36c10af0e59b8304de72127b4d0406004874f5
|
|
| MD5 |
35592bd6aed4547419c1e241c4b90c9e
|
|
| BLAKE2b-256 |
d100f895586b7f06fa297724ae0fc6715c4cc2a5919e183c6ee5099ccdc81896
|
Provenance
The following attestation bundles were made for autoray-0.10.1.tar.gz:
Publisher:
pypi-release.yml on jcmgray/autoray
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
autoray-0.10.1.tar.gz -
Subject digest:
b158904f410757f326d825e57b36c10af0e59b8304de72127b4d0406004874f5 - Sigstore transparency entry: 2373370033
- Sigstore integration time:
-
Permalink:
jcmgray/autoray@a0392d6e411bb06769fb1f98d625fe7badcbbb7c -
Branch / Tag:
refs/tags/v0.10.1 - Owner: https://github.com/jcmgray
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
pypi-release.yml@a0392d6e411bb06769fb1f98d625fe7badcbbb7c -
Trigger Event:
release
-
Statement type:
File details
Details for the file autoray-0.10.1-py3-none-any.whl.
File metadata
- Download URL: autoray-0.10.1-py3-none-any.whl
- Upload date:
- Size: 951.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
twine/6.1.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1e64918231fd0a166c90b8150d488c4d4e9f9e275d66518fc90bb221361fe6fd
|
|
| MD5 |
43284ca7b9ae0b20ebf5f16f2bc9334c
|
|
| BLAKE2b-256 |
373a7e233b8335f56fcc5fe06260d472d3eb68d82561fd96b8ad88517e03b0f8
|
Provenance
The following attestation bundles were made for autoray-0.10.1-py3-none-any.whl:
Publisher:
pypi-release.yml on jcmgray/autoray
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
autoray-0.10.1-py3-none-any.whl -
Subject digest:
1e64918231fd0a166c90b8150d488c4d4e9f9e275d66518fc90bb221361fe6fd - Sigstore transparency entry: 2373370630
- Sigstore integration time:
-
Permalink:
jcmgray/autoray@a0392d6e411bb06769fb1f98d625fe7badcbbb7c -
Branch / Tag:
refs/tags/v0.10.1 - Owner: https://github.com/jcmgray
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
pypi-release.yml@a0392d6e411bb06769fb1f98d625fe7badcbbb7c -
Trigger Event:
release
-
Statement type:
