Array API Standard Support: special#

Support on CPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

torch

jax

dask

bdtr

✔️

✔️

✔️

bdtrc

✔️

✔️

✔️

bdtri

✔️

✔️

✔️

bdtrik

bdtrin

betainc

✔️

✔️

✔️

betaincc

✔️

✔️

✔️

betainccinv

betaincinv

✔️

✔️

✔️

btdtria

btdtrib

chdtriv

chndtr

chndtridf

chndtrinc

chndtrix

elliprc

elliprd

elliprf

elliprg

elliprj

erfinv

✔️

✔️

✔️

eval_chebyc

eval_chebys

eval_chebyt

eval_chebyu

eval_gegenbauer

eval_genlaguerre

eval_hermite

eval_hermitenorm

eval_jacobi

eval_laguerre

eval_legendre

eval_sh_chebyt

eval_sh_chebyu

eval_sh_jacobi

eval_sh_legendre

expn

✔️

✔️

✔️

fdtr

✔️

✔️

✔️

fdtrc

✔️

✔️

✔️

fdtri

✔️

✔️

✔️

fdtridfd

hyp1f1

✔️

✔️

✔️

kn

log_gammainc

log_gammaincc

nbdtr

✔️

✔️

✔️

nbdtrc

✔️

✔️

✔️

nbdtri

✔️

✔️

✔️

nbdtrik

nbdtrin

ncfdtr

ncfdtri

ncfdtridfd

ncfdtridfn

ncfdtrinc

nctdtr

nctdtridf

nctdtrinc

nctdtrit

pdtri

✔️

✔️

✔️

pdtrik

powm1

shichi

sici

smirnov

smirnovi

stdtr

✔️

✔️

✔️

stdtridf

stdtrit

✔️

✖

✔️

yn

✔️

✔️

✔️

geterr

seterr

errstate

agm

airy

airye

bei

beip

ber

berp

besselpoly

beta

betaln

✔️

✔️

✔️

binom

✔️

✔️

✔️

boxcox

✔️

✔️

✔️

boxcox1p

✔️

✔️

✔️

cbrt

✔️

✔️

✔️

chdtr

✔️

✔️

✔️

chdtrc

✔️

✔️

✔️

chdtri

✔️

✔️

✔️

cosdg

✔️

✔️

✔️

cosm1

✔️

✔️

✔️

cotdg

✔️

✔️

✔️

dawsn

digammainv

ellipe

ellipeinc

ellipj

ellipk

✔️

✔️

✔️

ellipkinc

ellipkm1

✔️

✔️

✔️

entr

✔️

✔️

✔️

erf

✔️

✔️

✔️

erfc

✔️

✔️

✔️

erfcinv

erfcx

✔️

✔️

✔️

erfi

exp1

✔️

✔️

✔️

exp10

✔️

✔️

✔️

exp2

✔️

✔️

✔️

expi

✔️

✔️

✔️

expit

✔️

✔️

✔️

expm1

exprel

✔️

✔️

✔️

fresnel

gamma

✔️

✔️

✔️

gammainc

✔️

✔️

✔️

gammaincc

✔️

✔️

✔️

gammainccinv

✔️

✔️

✔️

gammaincinv

✔️

✔️

✔️

gammaln

✔️

✔️

✔️

gammasgn

✔️

✔️

✔️

gdtr

✔️

✔️

✔️

gdtrc

✔️

✔️

✔️

gdtria

gdtrib

gdtrix

hankel1

hankel1e

hankel2

hankel2e

huber

✔️

✔️

✔️

hyp0f1

hyp2f1

✔️

✔️

✔️

hyperu

i0

✔️

✔️

✔️

i0e

✔️

✔️

✔️

i1

✔️

✔️

✔️

i1e

✔️

✔️

✔️

inv_boxcox

✔️

✔️

✔️

inv_boxcox1p

✔️

✔️

✔️

it2i0k0

it2j0y0

it2struve0

itairy

iti0k0

itj0y0

itmodstruve0

itstruve0

iv

ive

j0

✔️

✔️

✔️

j1

✔️

✔️

✔️

jv

jve

k0

✔️

✔️

✔️

k0e

✔️

✔️

✔️

k1

✔️

✔️

✔️

k1e

✔️

✔️

✔️

kei

keip

kelvin

ker

kerp

kl_div

✔️

✔️

✔️

kolmogi

kolmogorov

kv

kve

log1p

log_expit

log_ndtr

✔️

✔️

✔️

log_wright_bessel

loggamma

✔️

✔️

✔️

logit

✔️

✔️

✔️

lpmv

✔️

✔️

✔️

mathieu_a

mathieu_b

mathieu_cem

mathieu_modcem1

mathieu_modcem2

mathieu_modsem1

mathieu_modsem2

mathieu_sem

modfresnelm

modfresnelp

modstruve

ndtr

✔️

✔️

✔️

ndtri

✔️

✔️

✔️

ndtri_exp

nrdtrimn

nrdtrisd

obl_ang1

obl_ang1_cv

obl_cv

obl_rad1

obl_rad1_cv

obl_rad2

obl_rad2_cv

owens_t

pdtr

✔️

✔️

✔️

pdtrc

✔️

✔️

✔️

pbdv

pbvv

pbwa

poch

✔️

✔️

✔️

pro_ang1

pro_ang1_cv

pro_cv

pro_rad1

pro_rad1_cv

pro_rad2

pro_rad2_cv

pseudo_huber

✔️

✔️

✔️

psi

✔️

✔️

✔️

radian

✔️

✔️

✔️

rel_entr

✔️

✔️

✔️

rgamma

✔️

✔️

✔️

round

sindg

✔️

✔️

✔️

spence

✔️

✔️

✔️

struve

tandg

✔️

✔️

✔️

tklmbda

voigt_profile

wofz

wright_bessel

wrightomega

xlog1py

✔️

✔️

✔️

xlogy

✔️

✔️

✔️

y0

✔️

✔️

✔️

y1

✔️

✔️

✔️

yv

yve

zetac

✔️

✔️

✔️

ai_zeros

assoc_laguerre

bei_zeros

beip_zeros

ber_zeros

bernoulli

berp_zeros

bi_zeros

comb

digamma

✔️

✔️

✔️

diric

erf_zeros

euler

factorial

factorial2

factorialk

fresnel_zeros

fresnelc_zeros

fresnels_zeros

h1vp

h2vp

ivp

jn_zeros

jnjnp_zeros

jnp_zeros

jnyn_zeros

jvp

kei_zeros

keip_zeros

kelvin_zeros

ker_zeros

kerp_zeros

kvp

lmbda

lqmn

lqn

mathieu_even_coef

mathieu_odd_coef

obl_cv_seq

pbdn_seq

pbdv_seq

pbvv_seq

perm

polygamma

✔️

✔️

✔️

pro_cv_seq

riccati_jn

riccati_yn

sinc

✔️

✔️

✔️

softplus

stirling2

y0_zeros

y1_zeros

y1p_zeros

yn_zeros

ynp_zeros

yvp

zeta

✔️

✔️

✔️

legendre

chebyt

chebyu

chebyc

chebys

jacobi

laguerre

genlaguerre

hermite

hermitenorm

gegenbauer

sh_legendre

sh_chebyt

sh_chebyu

sh_jacobi

roots_legendre

roots_chebyt

roots_chebyu

roots_chebyc

roots_chebys

roots_jacobi

roots_laguerre

roots_genlaguerre

roots_hermite

roots_hermitenorm

roots_gegenbauer

roots_sh_legendre

roots_sh_chebyt

roots_sh_chebyu

roots_sh_jacobi

assoc_legendre_p

assoc_legendre_p_all

legendre_p

legendre_p_all

sph_harm_y

sph_harm_y_all

sph_legendre_p

sph_legendre_p_all

logsumexp

✔️

✔️

✔️

softmax

✔️

✔️

✔️

log_softmax

✔️

✔️

✔️

multigammaln

✔️

✔️

✔️

ellip_harm

ellip_harm_2

ellip_normal

lambertw

spherical_jn

spherical_yn

spherical_in

spherical_kn

Support on GPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

cupy

torch

jax

bdtr

✔️

✖

✖

bdtrc

✔️

✖

✖

bdtri

✔️

✖

✖

bdtrik

bdtrin

betainc

✔️

✖

✔️

betaincc

✔️

✖

✔️

betainccinv

betaincinv

✔️

✖

✖

btdtria

btdtrib

chdtriv

chndtr

chndtridf

chndtrinc

chndtrix

elliprc

elliprd

elliprf

elliprg

elliprj

erfinv

✔️

✔️

✔️

eval_chebyc

eval_chebys

eval_chebyt

eval_chebyu

eval_gegenbauer

eval_genlaguerre

eval_hermite

eval_hermitenorm

eval_jacobi

eval_laguerre

eval_legendre

eval_sh_chebyt

eval_sh_chebyu

eval_sh_jacobi

eval_sh_legendre

expn

✔️

✖

✔️

fdtr

✔️

✖

✖

fdtrc

✔️

✖

✖

fdtri

✔️

✖

✖

fdtridfd

hyp1f1

✖

✖

✔️

kn

log_gammainc

log_gammaincc

nbdtr

✔️

✖

✖

nbdtrc

✔️

✖

✖

nbdtri

✔️

✖

✖

nbdtrik

nbdtrin

ncfdtr

ncfdtri

ncfdtridfd

ncfdtridfn

ncfdtrinc

nctdtr

nctdtridf

nctdtrinc

nctdtrit

pdtri

✔️

✖

✖

pdtrik

powm1

shichi

sici

smirnov

smirnovi

stdtr

✔️

✖

✔️

stdtridf

stdtrit

✔️

✖

✖

yn

✔️

✖

✖

geterr

seterr

errstate

agm

airy

airye

bei

beip

ber

berp

besselpoly

beta

betaln

✔️

✖

✔️

binom

✔️

✖

✖

boxcox

✔️

✖

✖

boxcox1p

✔️

✖

✖

cbrt

✔️

✖

✖

chdtr

✔️

✔️

✔️

chdtrc

✔️

✔️

✔️

chdtri

✔️

✖

✖

cosdg

✔️

✖

✖

cosm1

✔️

✖

✖

cotdg

✔️

✖

✖

dawsn

digammainv

ellipe

ellipeinc

ellipj

ellipk

✔️

✖

✖

ellipkinc

ellipkm1

✔️

✖

✖

entr

✔️

✔️

✔️

erf

✔️

✔️

✔️

erfc

✔️

✔️

✔️

erfcinv

erfcx

✔️

✔️

✖

erfi

exp1

✔️

✖

✖

exp10

✔️

✖

✖

exp2

✔️

✖

✖

expi

✔️

✖

✔️

expit

✔️

✔️

✔️

expm1

exprel

✔️

✖

✖

fresnel

gamma

✔️

✖

✔️

gammainc

✔️

✔️

✔️

gammaincc

✔️

✔️

✔️

gammainccinv

✔️

✖

✖

gammaincinv

✔️

✖

✖

gammaln

✔️

✔️

✔️

gammasgn

✔️

✖

✔️

gdtr

✔️

✖

✖

gdtrc

✔️

✖

✖

gdtria

gdtrib

gdtrix

hankel1

hankel1e

hankel2

hankel2e

huber

✔️

✖

✖

hyp0f1

hyp2f1

✖

✖

✔️

hyperu

i0

✔️

✔️

✔️

i0e

✔️

✔️

✔️

i1

✔️

✔️

✔️

i1e

✔️

✔️

✔️

inv_boxcox

✔️

✖

✖

inv_boxcox1p

✔️

✖

✖

it2i0k0

it2j0y0

it2struve0

itairy

iti0k0

itj0y0

itmodstruve0

itstruve0

iv

ive

j0

✔️

✔️

✖

j1

✔️

✔️

✖

jv

jve

k0

✔️

✔️

✖

k0e

✔️

✔️

✖

k1

✔️

✔️

✖

k1e

✔️

✔️

✖

kei

keip

kelvin

ker

kerp

kl_div

✔️

✖

✔️

kolmogi

kolmogorov

kv

kve

log1p

log_expit

log_ndtr

✔️

✔️

✔️

log_wright_bessel

loggamma

✔️

✖

✖

logit

✔️

✔️

✔️

lpmv

✔️

✖

✖

mathieu_a

mathieu_b

mathieu_cem

mathieu_modcem1

mathieu_modcem2

mathieu_modsem1

mathieu_modsem2

mathieu_sem

modfresnelm

modfresnelp

modstruve

ndtr

✔️

✔️

✔️

ndtri

✔️

✔️

✔️

ndtri_exp

nrdtrimn

nrdtrisd

obl_ang1

obl_ang1_cv

obl_cv

obl_rad1

obl_rad1_cv

obl_rad2

obl_rad2_cv

owens_t

pdtr

✔️

✖

✖

pdtrc

✔️

✖

✖

pbdv

pbvv

pbwa

poch

✔️

✖

✔️

pro_ang1

pro_ang1_cv

pro_cv

pro_rad1

pro_rad1_cv

pro_rad2

pro_rad2_cv

pseudo_huber

✔️

✖

✖

psi

✔️

✔️

✔️

radian

✔️

✖

✖

rel_entr

✔️

✔️

✔️

rgamma

✔️

✖

✖

round

sindg

✔️

✖

✖

spence

✖

✖

✔️

struve

tandg

✔️

✖

✖

tklmbda

voigt_profile

wofz

wright_bessel

wrightomega

xlog1py

✔️

✔️

✔️

xlogy

✔️

✔️

✔️

y0

✔️

✔️

✖

y1

✔️

✔️

✖

yv

yve

zetac

✔️

✖

✖

ai_zeros

assoc_laguerre

bei_zeros

beip_zeros

ber_zeros

bernoulli

berp_zeros

bi_zeros

comb

digamma

✔️

✔️

✔️

diric

erf_zeros

euler

factorial

factorial2

factorialk

fresnel_zeros

fresnelc_zeros

fresnels_zeros

h1vp

h2vp

ivp

jn_zeros

jnjnp_zeros

jnp_zeros

jnyn_zeros

jvp

kei_zeros

keip_zeros

kelvin_zeros

ker_zeros

kerp_zeros

kvp

lmbda

lqmn

lqn

mathieu_even_coef

mathieu_odd_coef

obl_cv_seq

pbdn_seq

pbdv_seq

pbvv_seq

perm

polygamma

✔️

✔️

✔️

pro_cv_seq

riccati_jn

riccati_yn

sinc

✔️

✔️

✖

softplus

stirling2

y0_zeros

y1_zeros

y1p_zeros

yn_zeros

ynp_zeros

yvp

zeta

✔️

✔️

✔️

legendre

chebyt

chebyu

chebyc

chebys

jacobi

laguerre

genlaguerre

hermite

hermitenorm

gegenbauer

sh_legendre

sh_chebyt

sh_chebyu

sh_jacobi

roots_legendre

roots_chebyt

roots_chebyu

roots_chebyc

roots_chebys

roots_jacobi

roots_laguerre

roots_genlaguerre

roots_hermite

roots_hermitenorm

roots_gegenbauer

roots_sh_legendre

roots_sh_chebyt

roots_sh_chebyu

roots_sh_jacobi

assoc_legendre_p

assoc_legendre_p_all

legendre_p

legendre_p_all

sph_harm_y

sph_harm_y_all

sph_legendre_p

sph_legendre_p_all

logsumexp

✔️

✔️

✔️

softmax

✔️

✔️

✔️

log_softmax

✔️

✔️

✔️

multigammaln

✔️

✔️

✔️

ellip_harm

ellip_harm_2

ellip_normal

lambertw

spherical_jn

spherical_yn

spherical_in

spherical_kn

Support with JIT#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

jax

bdtr

✔️

bdtrc

✔️

bdtri

✔️

bdtrik

bdtrin

betainc

✔️

betaincc

✔️

betainccinv

betaincinv

✔️

btdtria

btdtrib

chdtriv

chndtr

chndtridf

chndtrinc

chndtrix

elliprc

elliprd

elliprf

elliprg

elliprj

erfinv

✔️

eval_chebyc

eval_chebys

eval_chebyt

eval_chebyu

eval_gegenbauer

eval_genlaguerre

eval_hermite

eval_hermitenorm

eval_jacobi

eval_laguerre

eval_legendre

eval_sh_chebyt

eval_sh_chebyu

eval_sh_jacobi

eval_sh_legendre

expn

✔️

fdtr

✔️

fdtrc

✔️

fdtri

✔️

fdtridfd

hyp1f1

✔️

kn

log_gammainc

log_gammaincc

nbdtr

✔️

nbdtrc

✔️

nbdtri

✔️

nbdtrik

nbdtrin

ncfdtr

ncfdtri

ncfdtridfd

ncfdtridfn

ncfdtrinc

nctdtr

nctdtridf

nctdtrinc

nctdtrit

pdtri

✔️

pdtrik

powm1

shichi

sici

smirnov

smirnovi

stdtr

✔️

stdtridf

stdtrit

✖

yn

✔️

geterr

seterr

errstate

agm

airy

airye

bei

beip

ber

berp

besselpoly

beta

betaln

✔️

binom

✔️

boxcox

✔️

boxcox1p

✔️

cbrt

✔️

chdtr

✔️

chdtrc

✔️

chdtri

✔️

cosdg

✔️

cosm1

✔️

cotdg

✔️

dawsn

digammainv

ellipe

ellipeinc

ellipj

ellipk

✔️

ellipkinc

ellipkm1

✔️

entr

✔️

erf

✔️

erfc

✔️

erfcinv

erfcx

✔️

erfi

exp1

✔️

exp10

✔️

exp2

✔️

expi

✔️

expit

✔️

expm1

exprel

✔️

fresnel

gamma

✔️

gammainc

✔️

gammaincc

✔️

gammainccinv

✔️

gammaincinv

✔️

gammaln

✔️

gammasgn

✔️

gdtr

✔️

gdtrc

✔️

gdtria

gdtrib

gdtrix

hankel1

hankel1e

hankel2

hankel2e

huber

✔️

hyp0f1

hyp2f1

✔️

hyperu

i0

✔️

i0e

✔️

i1

✔️

i1e

✔️

inv_boxcox

✔️

inv_boxcox1p

✔️

it2i0k0

it2j0y0

it2struve0

itairy

iti0k0

itj0y0

itmodstruve0

itstruve0

iv

ive

j0

✔️

j1

✔️

jv

jve

k0

✔️

k0e

✔️

k1

✔️

k1e

✔️

kei

keip

kelvin

ker

kerp

kl_div

✔️

kolmogi

kolmogorov

kv

kve

log1p

log_expit

log_ndtr

✔️

log_wright_bessel

loggamma

✔️

logit

✔️

lpmv

✔️

mathieu_a

mathieu_b

mathieu_cem

mathieu_modcem1

mathieu_modcem2

mathieu_modsem1

mathieu_modsem2

mathieu_sem

modfresnelm

modfresnelp

modstruve

ndtr

✔️

ndtri

✔️

ndtri_exp

nrdtrimn

nrdtrisd

obl_ang1

obl_ang1_cv

obl_cv

obl_rad1

obl_rad1_cv

obl_rad2

obl_rad2_cv

owens_t

pdtr

✔️

pdtrc

✔️

pbdv

pbvv

pbwa

poch

✔️

pro_ang1

pro_ang1_cv

pro_cv

pro_rad1

pro_rad1_cv

pro_rad2

pro_rad2_cv

pseudo_huber

✔️

psi

✔️

radian

✔️

rel_entr

✔️

rgamma

✔️

round

sindg

✔️

spence

✔️

struve

tandg

✔️

tklmbda

voigt_profile

wofz

wright_bessel

wrightomega

xlog1py

✔️

xlogy

✔️

y0

✔️

y1

✔️

yv

yve

zetac

✔️

ai_zeros

assoc_laguerre

bei_zeros

beip_zeros

ber_zeros

bernoulli

berp_zeros

bi_zeros

comb

digamma

✔️

diric

erf_zeros

euler

factorial

factorial2

factorialk

fresnel_zeros

fresnelc_zeros

fresnels_zeros

h1vp

h2vp

ivp

jn_zeros

jnjnp_zeros

jnp_zeros

jnyn_zeros

jvp

kei_zeros

keip_zeros

kelvin_zeros

ker_zeros

kerp_zeros

kvp

lmbda

lqmn

lqn

mathieu_even_coef

mathieu_odd_coef

obl_cv_seq

pbdn_seq

pbdv_seq

pbvv_seq

perm

polygamma

✔️

pro_cv_seq

riccati_jn

riccati_yn

sinc

✔️

softplus

stirling2

y0_zeros

y1_zeros

y1p_zeros

yn_zeros

ynp_zeros

yvp

zeta

✔️

legendre

chebyt

chebyu

chebyc

chebys

jacobi

laguerre

genlaguerre

hermite

hermitenorm

gegenbauer

sh_legendre

sh_chebyt

sh_chebyu

sh_jacobi

roots_legendre

roots_chebyt

roots_chebyu

roots_chebyc

roots_chebys

roots_jacobi

roots_laguerre

roots_genlaguerre

roots_hermite

roots_hermitenorm

roots_gegenbauer

roots_sh_legendre

roots_sh_chebyt

roots_sh_chebyu

roots_sh_jacobi

assoc_legendre_p

assoc_legendre_p_all

legendre_p

legendre_p_all

sph_harm_y

sph_harm_y_all

sph_legendre_p

sph_legendre_p_all

logsumexp

✔️

softmax

✔️

log_softmax

✔️

multigammaln

✔️

ellip_harm

ellip_harm_2

ellip_normal

lambertw

spherical_jn

spherical_yn

spherical_in

spherical_kn