Projects Involved

Array-API

uarray

SciPy

NumPy

CuPy

PyTorch

What's Interoperability?

The ability of computer systems or software to exchange and make use of information.

Array-API Standard Goals?

What is SciPy?

All things "Sciency" put together.

What is uarray?

• backend / dispatch mechanism
• separately define an API, along with backends containing separate implementations of that API.
• user can register an implementation

SciPy | uarray

# SciPy ndimage with CuPy array
from scipy import ndimage
import cupy as cp

with scipy.ndimage.set_backend('cupy'):
    y_cupy = ndimage.correlate1d(cp.arange(10),
                                 cp.array([1, 2.5]))

SciPy | Array API

from scipy import signal
import torch

fs = 10e3
x = torch.tensor(...)
f, Pxx_den = signal.welch(x, fs, nperseg=1024)

PyTorch

Improve Array-API Support (working currently)

Thanks!