Array API Standard Support: fft#

Support on CPU#

Legend

âœ”ī¸ = supported

✖ = unsupported

N/A = out-of-scope

function/class

torch

jax

dask

fft

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifft

âœ”ī¸

âœ”ī¸

âœ”ī¸

fft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

fftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

irfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

irfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

irfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

hfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

ihfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

hfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

ihfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

hfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

ihfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

fftfreq

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfftfreq

âœ”ī¸

âœ”ī¸

âœ”ī¸

fftshift

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifftshift

âœ”ī¸

âœ”ī¸

âœ”ī¸

next_fast_len

N/A

N/A

N/A

prev_fast_len

N/A

N/A

N/A

dct

âœ”ī¸

✖

âœ”ī¸

idct

âœ”ī¸

âœ”ī¸

âœ”ī¸

dst

âœ”ī¸

✖

âœ”ī¸

idst

âœ”ī¸

âœ”ī¸

âœ”ī¸

dctn

âœ”ī¸

âœ”ī¸

âœ”ī¸

idctn

âœ”ī¸

âœ”ī¸

âœ”ī¸

dstn

âœ”ī¸

âœ”ī¸

âœ”ī¸

idstn

âœ”ī¸

âœ”ī¸

âœ”ī¸

fht

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifht

âœ”ī¸

âœ”ī¸

âœ”ī¸

fhtoffset

N/A

N/A

N/A

set_backend

N/A

N/A

N/A

skip_backend

N/A

N/A

N/A

set_global_backend

N/A

N/A

N/A

register_backend

N/A

N/A

N/A

get_workers

N/A

N/A

N/A

set_workers

N/A

N/A

N/A

Support on GPU#

Legend

âœ”ī¸ = supported

✖ = unsupported

N/A = out-of-scope

function/class

cupy

torch

jax

fft

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifft

âœ”ī¸

âœ”ī¸

âœ”ī¸

fft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

fftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

irfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

irfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

irfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

hfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

ihfft

âœ”ī¸

âœ”ī¸

âœ”ī¸

hfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

ihfft2

âœ”ī¸

âœ”ī¸

âœ”ī¸

hfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

ihfftn

âœ”ī¸

âœ”ī¸

âœ”ī¸

fftfreq

âœ”ī¸

âœ”ī¸

âœ”ī¸

rfftfreq

âœ”ī¸

âœ”ī¸

âœ”ī¸

fftshift

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifftshift

âœ”ī¸

âœ”ī¸

âœ”ī¸

next_fast_len

N/A

N/A

N/A

prev_fast_len

N/A

N/A

N/A

dct

✖

✖

✖

idct

✖

✖

✖

dst

✖

✖

✖

idst

✖

✖

✖

dctn

✖

✖

✖

idctn

✖

✖

✖

dstn

✖

✖

✖

idstn

✖

✖

✖

fht

âœ”ī¸

âœ”ī¸

âœ”ī¸

ifht

âœ”ī¸

âœ”ī¸

âœ”ī¸

fhtoffset

N/A

N/A

N/A

set_backend

N/A

N/A

N/A

skip_backend

N/A

N/A

N/A

set_global_backend

N/A

N/A

N/A

register_backend

N/A

N/A

N/A

get_workers

N/A

N/A

N/A

set_workers

N/A

N/A

N/A

Support with JIT#

Legend

âœ”ī¸ = supported

✖ = unsupported

N/A = out-of-scope

function/class

jax

fft

âœ”ī¸

ifft

âœ”ī¸

fft2

âœ”ī¸

ifft2

âœ”ī¸

fftn

âœ”ī¸

ifftn

âœ”ī¸

rfft

âœ”ī¸

irfft

âœ”ī¸

rfft2

âœ”ī¸

irfft2

âœ”ī¸

rfftn

âœ”ī¸

irfftn

âœ”ī¸

hfft

âœ”ī¸

ihfft

âœ”ī¸

hfft2

âœ”ī¸

ihfft2

âœ”ī¸

hfftn

âœ”ī¸

ihfftn

âœ”ī¸

fftfreq

âœ”ī¸

rfftfreq

âœ”ī¸

fftshift

âœ”ī¸

ifftshift

âœ”ī¸

next_fast_len

N/A

prev_fast_len

N/A

dct

✖

idct

âœ”ī¸

dst

✖

idst

âœ”ī¸

dctn

âœ”ī¸

idctn

âœ”ī¸

dstn

âœ”ī¸

idstn

âœ”ī¸

fht

âœ”ī¸

ifht

âœ”ī¸

fhtoffset

N/A

set_backend

N/A

skip_backend

N/A

set_global_backend

N/A

register_backend

N/A

get_workers

N/A

set_workers

N/A