9 Commits

Author	SHA1	Message	Date
Jake VanderPlas	43a9faa06a	Rename _wraps to implements	2024-01-24 14:14:19 -08: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
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
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