5 Commits

Author	SHA1	Message	Date
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
Peter Hawkins	335b2cfb26	[JAX] Prepare not to export jax._src by default. ... Currently ``` import jax ``` populates `jax._src` in the names exported from JAX. This change prepares for not exporting `jax._src` by default. In particular, explicitly import modules from jax._src and refer to those imports rather than assuming jax._src contents will be around later. This is a common pattern in tests. This change does not yet remove any exported names. Issue https://github.com/google/jax/issues/11951 PiperOrigin-RevId: 469480816	2022-08-23 09:36:47 -07:00
Peter Hawkins	db2e91eba2	Move jax.test_util to jax._src.test_util. ... Add forwarding shims for names used by external clients of JAX in practice. PiperOrigin-RevId: 398721725	2021-09-24 07:02:49 -07:00
Peter Hawkins	2c2f4033cc	Move contents of jax.lib to jax._src.lib. ... Add shim libraries for functions exported from jax.lib that other code seems to use in practice. PiperOrigin-RevId: 398471863	2021-09-23 06:33:55 -07:00
Qiao Zhang	57ce886de2	[JAX] Skip deleted buffer and executable for heap profile. ... PiperOrigin-RevId: 389724193	2021-08-09 14:24:52 -07:00