Jake VanderPlas
43a9faa06a
Rename _wraps to implements
2024-01-24 14:14:19 -08:00
Jake VanderPlas
cf11f8da8a
stats.norm: add logsf & make sf more accurate near zero
2023-08-21 16:48:39 -07:00
Misha
83b3f5b759
Fix loc and scale parameters in scipy.logistic. Add CDF and SF for several distributions.
2023-03-21 00:16:13 +01:00
Jake VanderPlas
760deb310e
Remove leading underscores in jax._src.numpy.util
2023-03-13 12:18:36 -07:00
Jake VanderPlas
c8c269f5f5
internal: avoid unused imports in lax_numpy
2023-03-08 10:29:04 -08:00
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
Lukas Geiger
3a2e80ef51
Replace pow() with srqt() or square() where possible
2021-05-24 10:43:35 +01:00
Jake VanderPlas
f74235cdae
X32 tests: fail on dtype warnings
2020-12-08 13:03:30 -08:00
Jake VanderPlas
9d2f6148ed
Call asarray() rather than array() to avoid host round-trips.
2020-11-24 16:05:48 -08: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
