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
Misha
feb9ab33af
Fixed loc and scale parameters for logistic distribution. CDF and SF have been added for several distributions, including cauchy, gamma, logistic, chi2 and beta. ISF and PPF have also been added for cauchy and logistic.
2023-03-07 07:56:47 +01:00
Jake VanderPlas
b1119be830
[typing] annotage jax.scipy.stats distributions
2022-10-12 13:42:11 -07:00
Peter Hawkins
ba557d5e1b
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
...
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
2022-09-22 12:27:19 -07:00
Roy Frostig
f7731bf959
remove _const from public jax.lax module
...
Modify all internal call sites to use `jax._src.lax.lax._const`.
2022-03-07 12:26:25 -08:00
Jake VanderPlas
00e040e514
cleanup: remove _constant_like in favor of lax._const
2022-03-02 09:13:58 -08:00
Jake VanderPlas
743a1c270d
scipy.stats.beta: fix pdf for x=0, 1
2021-08-17 09:52:11 -07:00
Peter Hawkins
aa107cf1f4
Move jax.numpy internals into jax._src.numpy.
2020-10-16 20:35:19 -04:00
Peter Hawkins
6acb46516e
Move most of the implementation of jax.scipy into jax._src.scipy.
2020-10-16 17:04:25 -04:00
