Array API Standard Support: stats

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class	torch	jax	dask
Binomial
BootstrapMethod
CensoredData
Covariance
Logistic
Mixture
MonteCarloMethod
Normal
PermutationMethod
Uniform
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	✔️	✔️	✖
estimated_cdf	✔️	✔️	✖
exp	✖	✖	✖
expectile	✔️	✖	✖
f_oneway	✔️	✔️	✔️
false_discovery_control	✔️	✔️	✖
fisher_exact	✖	✖	✖
fit	N/A	N/A	N/A
fligner	✔️	✔️	✖
friedmanchisquare	✔️	✔️	✖
gaussian_kde
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	✔️	✔️	✖
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	N/A	N/A	N/A
relfreq	✔️	✔️	✖
rv_continuous
rv_discrete
rv_histogram
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

blank = not currently documented

function/class	cupy	torch	jax
Binomial
BootstrapMethod
CensoredData
Covariance
Logistic
Mixture
MonteCarloMethod
Normal
PermutationMethod
Uniform
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	✔️	✔️	✔️
estimated_cdf	✔️	✔️	✔️
exp	✖	✖	✖
expectile	✔️	✔️	✖
f_oneway	✔️	✖	✖
false_discovery_control	✔️	✔️	✔️
fisher_exact	✖	✖	✖
fit	N/A	N/A	N/A
fligner	✔️	✔️	✔️
friedmanchisquare	✔️	✔️	✔️
gaussian_kde
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	✖	✖	✖
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	N/A	N/A	N/A
relfreq	✔️	✔️	✔️
rv_continuous
rv_discrete
rv_histogram
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

blank = not currently documented

function/class	jax
Binomial
BootstrapMethod
CensoredData
Covariance
Logistic
Mixture
MonteCarloMethod
Normal
PermutationMethod
Uniform
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	✔️
estimated_cdf	✔️
exp	✖
expectile	✖
f_oneway	✔️
false_discovery_control	✔️
fisher_exact	✖
fit	N/A
fligner	✔️
friedmanchisquare	✔️
gaussian_kde
gmean	✔️
goodness_of_fit	N/A
gstd	✔️
gzscore	✔️
hmean	✔️
iqr	✔️
jarque_bera	✔️
kendalltau	✖
kruskal	✔️
ks_1samp	✖
ks_2samp	✖
kstat	✔️
kstatvar	✔️
kstest	✖
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	N/A
relfreq	✔️
rv_continuous
rv_discrete
rv_histogram
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	✔️