Array API Standard Support: stats

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	torch	jax	dask
abs	✖	✖	✖
alexandergovern	✔️	✔️	✔️
anderson	✖	✖	✖
anderson_ksamp	✖	✖	✖
ansari	✔️	✔️	✖
barnard_exact	✖	✖	✖
bartlett	✔️	✔️	✔️
bayes_mvs	✖	✖	✖
binned_statistic	✖	✖	✖
binned_statistic_2d	✖	✖	✖
binned_statistic_dd	✖	✖	✖
binomtest	✖	✖	✖
bootstrap	✔️	✖	✖
boschloo_exact	✖	✖	✖
boxcox	✖	✖	✖
boxcox_llf	✔️	✔️	✔️
boxcox_normmax	✖	✖	✖
boxcox_normplot	✖	✖	✖
brunnermunzel	✔️	✖	✖
bws_test	✖	✖	✖
chatterjeexi	✔️	✔️	✖
chi2_contingency	✖	✖	✖
chisquare	✔️	✔️	✔️
circmean	✔️	✔️	✔️
circstd	✔️	✔️	✔️
circvar	✔️	✔️	✔️
combine_pvalues	✔️	✔️	✔️
cramervonmises	✔️	✔️	✖
cramervonmises_2samp	✔️	✔️	✖
cumfreq	✖	✖	✖
describe	✔️	✔️	✔️
differential_entropy	✔️	✔️	✔️
directional_stats	✔️	✔️	✔️
dunnett	✖	✖	✖
ecdf	✖	✖	✖
energy_distance	✖	✖	✖
entropy	✔️	✔️	✔️
epps_singleton_2samp	✔️	✖	✖
exp	✖	✖	✖
expectile	✖	✖	✖
f_oneway	✔️	✔️	✔️
false_discovery_control	✔️	✔️	✖
fisher_exact	✖	✖	✖
fit	N/A	N/A	N/A
fligner	✔️	✔️	✖
friedmanchisquare	✔️	✔️	✖
gmean	✔️	✔️	✔️
goodness_of_fit	N/A	N/A	N/A
gstd	✔️	✔️	✔️
gzscore	✔️	✔️	✔️
hmean	✔️	✔️	✔️
iqr	✔️	✖	✖
jarque_bera	✔️	✔️	✔️
kendalltau	✖	✖	✖
kruskal	✔️	✔️	✖
ks_1samp	✔️	✔️	✖
ks_2samp	✖	✖	✖
kstat	✔️	✔️	✔️
kstatvar	✔️	✔️	✔️
kstest	N/A	N/A	N/A
kurtosis	✔️	✔️	✔️
kurtosistest	✔️	✔️	✔️
levene	✔️	✔️	✔️
linregress	✖	✖	✖
lmoment	✔️	✔️	✖
log	✖	✖	✖
logrank	✖	✖	✖
make_distribution	✖	✖	✖
mannwhitneyu	✔️	✔️	✖
median_abs_deviation	✔️	✖	✖
median_test	✖	✖	✖
mode	✔️	✖	✖
moment	✔️	✔️	✔️
monte_carlo_test	✔️	✔️	✔️
mood	✔️	✔️	✖
multiscale_graphcorr	✖	✖	✖
mvsdist	✖	✖	✖
normaltest	✔️	✔️	✔️
obrientransform	✖	✖	✖
order_statistic	✖	✖	✖
page_trend_test	✖	✖	✖
pearsonr	✔️	✔️	✔️
percentileofscore	✖	✖	✖
permutation_test	✔️	✔️	✖
pmean	✔️	✔️	✔️
pointbiserialr	✖	✖	✖
poisson_means_test	✖	✖	✖
power	✔️	✔️	✔️
power_divergence	✔️	✔️	✔️
ppcc_max	✖	✖	✖
ppcc_plot	✖	✖	✖
probplot	✖	✖	✖
quantile	✔️	✔️	✖
quantile_test	✖	✖	✖
rankdata	✔️	✔️	✖
ranksums	✖	✖	✖
relfreq	✖	✖	✖
scoreatpercentile	✖	✖	✖
sem	✔️	✔️	✔️
shapiro	✖	✖	✖
siegelslopes	✖	✖	✖
sigmaclip	✔️	✔️	✖
skew	✔️	✔️	✔️
skewtest	✔️	✔️	✔️
sobol_indices	✖	✖	✖
somersd	✖	✖	✖
spearmanr	✖	✖	✖
spearmanrho	✔️	✔️	✖
theilslopes	✖	✖	✖
tiecorrect	✖	✖	✖
tmax	✔️	✔️	✔️
tmean	✔️	✔️	✔️
tmin	✔️	✔️	✔️
trim1	✖	✖	✖
trim_mean	✔️	✔️	✔️
trimboth	✖	✖	✖
truncate	✖	✖	✖
tsem	✔️	✔️	✔️
tstd	✔️	✔️	✔️
ttest_1samp	✔️	✔️	✔️
ttest_ind	✔️	✔️	✔️
ttest_ind_from_stats	✔️	✔️	✔️
ttest_rel	✔️	✔️	✔️
tukey_hsd	✖	✖	✖
tvar	✔️	✔️	✔️
variation	✔️	✔️	✔️
wasserstein_distance	✖	✖	✖
wasserstein_distance_nd	✖	✖	✖
weightedtau	✖	✖	✖
wilcoxon	✔️	✔️	✖
yeojohnson	✖	✖	✖
yeojohnson_llf	✔️	✔️	✖
yeojohnson_normmax	✖	✖	✖
yeojohnson_normplot	✖	✖	✖
zmap	✔️	✔️	✔️
zscore	✔️	✔️	✔️

Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	cupy	torch	jax
abs	✖	✖	✖
alexandergovern	✔️	✔️	✔️
anderson	✖	✖	✖
anderson_ksamp	✖	✖	✖
ansari	✖	✖	✖
barnard_exact	✖	✖	✖
bartlett	✔️	✔️	✔️
bayes_mvs	✖	✖	✖
binned_statistic	✖	✖	✖
binned_statistic_2d	✖	✖	✖
binned_statistic_dd	✖	✖	✖
binomtest	✖	✖	✖
bootstrap	✔️	✔️	✖
boschloo_exact	✖	✖	✖
boxcox	✖	✖	✖
boxcox_llf	✔️	✔️	✔️
boxcox_normmax	✖	✖	✖
boxcox_normplot	✖	✖	✖
brunnermunzel	✖	✖	✖
bws_test	✖	✖	✖
chatterjeexi	✖	✔️	✔️
chi2_contingency	✖	✖	✖
chisquare	✔️	✔️	✔️
circmean	✔️	✔️	✔️
circstd	✔️	✔️	✔️
circvar	✔️	✔️	✔️
combine_pvalues	✔️	✖	✔️
cramervonmises	✖	✖	✖
cramervonmises_2samp	✖	✖	✖
cumfreq	✖	✖	✖
describe	✔️	✔️	✔️
differential_entropy	✔️	✔️	✔️
directional_stats	✔️	✔️	✔️
dunnett	✖	✖	✖
ecdf	✖	✖	✖
energy_distance	✖	✖	✖
entropy	✔️	✔️	✔️
epps_singleton_2samp	✔️	✔️	✖
exp	✖	✖	✖
expectile	✖	✖	✖
f_oneway	✔️	✖	✖
false_discovery_control	✔️	✔️	✔️
fisher_exact	✖	✖	✖
fit	N/A	N/A	N/A
fligner	✖	✔️	✔️
friedmanchisquare	✖	✔️	✔️
gmean	✔️	✔️	✔️
goodness_of_fit	N/A	N/A	N/A
gstd	✔️	✔️	✔️
gzscore	✔️	✔️	✔️
hmean	✔️	✔️	✔️
iqr	✔️	✔️	✖
jarque_bera	✔️	✔️	✔️
kendalltau	✖	✖	✖
kruskal	✖	✔️	✔️
ks_1samp	✖	✖	✖
ks_2samp	✖	✖	✖
kstat	✔️	✔️	✔️
kstatvar	✔️	✔️	✔️
kstest	N/A	N/A	N/A
kurtosis	✔️	✔️	✔️
kurtosistest	✔️	✔️	✔️
levene	✔️	✖	✖
linregress	✖	✖	✖
lmoment	✔️	✖	✔️
log	✖	✖	✖
logrank	✖	✖	✖
make_distribution	✖	✖	✖
mannwhitneyu	✖	✖	✖
median_abs_deviation	✔️	✔️	✖
median_test	✖	✖	✖
mode	✖	✔️	✖
moment	✔️	✔️	✔️
monte_carlo_test	✔️	✔️	✔️
mood	✖	✔️	✔️
multiscale_graphcorr	✖	✖	✖
mvsdist	✖	✖	✖
normaltest	✔️	✔️	✔️
obrientransform	✖	✖	✖
order_statistic	✖	✖	✖
page_trend_test	✖	✖	✖
pearsonr	✔️	✖	✔️
percentileofscore	✖	✖	✖
permutation_test	✔️	✔️	✔️
pmean	✔️	✔️	✔️
pointbiserialr	✖	✖	✖
poisson_means_test	✖	✖	✖
power	✔️	✔️	✔️
power_divergence	✔️	✔️	✔️
ppcc_max	✖	✖	✖
ppcc_plot	✖	✖	✖
probplot	✖	✖	✖
quantile	✔️	✔️	✔️
quantile_test	✖	✖	✖
rankdata	✖	✔️	✔️
ranksums	✖	✖	✖
relfreq	✖	✖	✖
scoreatpercentile	✖	✖	✖
sem	✔️	✔️	✔️
shapiro	✖	✖	✖
siegelslopes	✖	✖	✖
sigmaclip	✔️	✔️	✔️
skew	✔️	✔️	✔️
skewtest	✔️	✔️	✔️
sobol_indices	✖	✖	✖
somersd	✖	✖	✖
spearmanr	✖	✖	✖
spearmanrho	✖	✖	✔️
theilslopes	✖	✖	✖
tiecorrect	✖	✖	✖
tmax	✔️	✔️	✔️
tmean	✔️	✔️	✔️
tmin	✔️	✔️	✔️
trim1	✖	✖	✖
trim_mean	✔️	✔️	✔️
trimboth	✖	✖	✖
truncate	✖	✖	✖
tsem	✔️	✔️	✔️
tstd	✔️	✔️	✔️
ttest_1samp	✔️	✖	✔️
ttest_ind	✔️	✖	✔️
ttest_ind_from_stats	✔️	✖	✔️
ttest_rel	✔️	✖	✔️
tukey_hsd	✖	✖	✖
tvar	✔️	✔️	✔️
variation	✔️	✔️	✔️
wasserstein_distance	✖	✖	✖
wasserstein_distance_nd	✖	✖	✖
weightedtau	✖	✖	✖
wilcoxon	✖	✖	✖
yeojohnson	✖	✖	✖
yeojohnson_llf	✔️	✔️	✔️
yeojohnson_normmax	✖	✖	✖
yeojohnson_normplot	✖	✖	✖
zmap	✔️	✔️	✔️
zscore	✔️	✔️	✔️

Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	jax
abs	✖
alexandergovern	✔️
anderson	✖
anderson_ksamp	✖
ansari	✖
barnard_exact	✖
bartlett	✔️
bayes_mvs	✖
binned_statistic	✖
binned_statistic_2d	✖
binned_statistic_dd	✖
binomtest	✖
bootstrap	✖
boschloo_exact	✖
boxcox	✖
boxcox_llf	✔️
boxcox_normmax	✖
boxcox_normplot	✖
brunnermunzel	✖
bws_test	✖
chatterjeexi	✔️
chi2_contingency	✖
chisquare	✖
circmean	✔️
circstd	✔️
circvar	✔️
combine_pvalues	✖
cramervonmises	✖
cramervonmises_2samp	✖
cumfreq	✖
describe	✖
differential_entropy	✔️
directional_stats	✔️
dunnett	✖
ecdf	✖
energy_distance	✖
entropy	✔️
epps_singleton_2samp	✖
exp	✖
expectile	✖
f_oneway	✖
false_discovery_control	✖
fisher_exact	✖
fit	N/A
fligner	✖
friedmanchisquare	✖
gmean	✔️
goodness_of_fit	N/A
gstd	✔️
gzscore	✔️
hmean	✖
iqr	✖
jarque_bera	✖
kendalltau	✖
kruskal	✖
ks_1samp	✖
ks_2samp	✖
kstat	✔️
kstatvar	✔️
kstest	N/A
kurtosis	✖
kurtosistest	✔️
levene	✔️
linregress	✖
lmoment	✖
log	✖
logrank	✖
make_distribution	✖
mannwhitneyu	✖
median_abs_deviation	✖
median_test	✖
mode	✖
moment	✖
monte_carlo_test	✔️
mood	✔️
multiscale_graphcorr	✖
mvsdist	✖
normaltest	✔️
obrientransform	✖
order_statistic	✖
page_trend_test	✖
pearsonr	✔️
percentileofscore	✖
permutation_test	✔️
pmean	✖
pointbiserialr	✖
poisson_means_test	✖
power	✖
power_divergence	✖
ppcc_max	✖
ppcc_plot	✖
probplot	✖
quantile	✖
quantile_test	✖
rankdata	✖
ranksums	✖
relfreq	✖
scoreatpercentile	✖
sem	✖
shapiro	✖
siegelslopes	✖
sigmaclip	✖
skew	✖
skewtest	✔️
sobol_indices	✖
somersd	✖
spearmanr	✖
spearmanrho	✔️
theilslopes	✖
tiecorrect	✖
tmax	✔️
tmean	✔️
tmin	✔️
trim1	✖
trim_mean	✔️
trimboth	✖
truncate	✖
tsem	✔️
tstd	✔️
ttest_1samp	✖
ttest_ind	✔️
ttest_ind_from_stats	✔️
ttest_rel	✖
tukey_hsd	✖
tvar	✔️
variation	✔️
wasserstein_distance	✖
wasserstein_distance_nd	✖
weightedtau	✖
wilcoxon	✖
yeojohnson	✖
yeojohnson_llf	✔️
yeojohnson_normmax	✖
yeojohnson_normplot	✖
zmap	✔️
zscore	✔️