Parker Schuh d62fc88fb1 Roll back #14792 ...

Breaks tests. lax.sub requires arguments to have the same dtypes, got float32, float64. (Tip: jnp.subtract is a similar function that does automatic type promotion on inputs).

PiperOrigin-RevId: 514897538

2023-03-07 18:31:19 -08:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

_core.py

Replace jax._src.util.prod with math.prod.

2023-02-28 12:41:00 -08:00

bernoulli.py

Added scipy.stats.bernoulli cdf and ppf.

2022-12-22 18:12:25 +00:00

beta.py

Roll back #14792

2023-03-07 18:31:19 -08:00

betabinom.py

Cleanup: remove unused imports in private modules

2022-10-20 14:37:21 -07:00

cauchy.py

Roll back #14792

2023-03-07 18:31:19 -08:00

chi2.py

Roll back #14792

2023-03-07 18:31:19 -08:00

dirichlet.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

expon.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

gamma.py

Roll back #14792

2023-03-07 18:31:19 -08:00

gennorm.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

geom.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

kde.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

laplace.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

logistic.py

Roll back #14792

2023-03-07 18:31:19 -08:00

multinomial.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

multivariate_normal.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

nbinom.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

norm.py

Roll back #14792

2023-03-07 18:31:19 -08:00

pareto.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

poisson.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

t.py

Roll back #14792

2023-03-07 18:31:19 -08:00

truncnorm.py

implement truncnorm in jax.scipy.stats

2022-10-22 15:48:20 -04:00

uniform.py

[typing] annotage jax.scipy.stats distributions

2022-10-12 13:42:11 -07:00

vonmises.py

fix typo

2023-03-02 13:58:18 +08:00