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
