Sergei Lebedev fd3a8b2cc6 Deprecated define_* and DEFINE_* methods on jax.config ...

These methods are internal to JAX. Yet, prior to this commit they were
effectively part of the public API, since users could (and some did!) invoke
them on `jax.config`.

2023-10-29 20:58:19 +00:00

..

_src

Deprecated define_* and DEFINE_* methods on jax.config

2023-10-29 20:58:19 +00:00

example_libraries

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

experimental

[JAX] Add an option subset_by_index that allows computing a contiguous subset of eigenvalues from eigh.

2023-10-27 09:29:33 -07:00

extend

[random] remove internal uses of deprecated prng.seed_with_impl()

2023-10-17 13:18:08 -07:00

image

Document why 'import name as name' is used

2022-12-14 15:07:04 -08:00

interpreters

jax.interpreters.ad: deprecate inadvertently exported symbols

2023-10-16 11:22:37 -07:00

lax

Use totalorder comparisons for sort

2023-10-13 12:21:07 -07:00

lib

Move compiler APIs out of dispatch.py and xla_bridge.py into a new jax._src.compiler module.

2023-08-15 06:39:46 -07:00

nn

Add truncated normal initializer to jax.nn

2023-09-07 16:23:42 -07:00

numpy

[CI] avoid referencing numpy.core to fix nightly CI

2023-10-23 09:11:33 -07:00

ops

Document why 'import name as name' is used

2022-12-14 15:07:04 -08:00

scipy

feat: add wrapcauchy logpdf and pdf

2023-10-18 13:47:10 +02:00

tools

Fix jaxlib build failure after upstream MLIR Python binding changes.

2023-10-23 14:27:52 +00:00

__init__.py

MAINT Do not use str() and repr() in f-string replacement fields

2023-10-23 15:12:04 +01:00

ad_checkpoint.py

Add memories support to remat.

2023-09-12 20:50:05 -07:00

api_util.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

BUILD

Set public module for exported jax.dtypes APIs

2023-10-17 15:07:28 -07:00

cloud_tpu_init.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

collect_profile.py

Cleanup: remove unused imports in private modules

2022-10-20 14:37:21 -07:00

config.py

Deprecated accessing config via the jax.config submodule

2023-10-27 20:08:10 +01:00

core.py

jax.core: point deprecation to jax.extend

2023-10-13 12:49:05 -07:00

custom_batching.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

custom_derivatives.py

custom_vjp symbolic zeros support, take two

2023-04-05 11:17:05 -07:00

custom_transpose.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

debug.py

Explicitly re-export all names from jax.debug

2023-04-07 07:24:59 -07:00

distributed.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

dlpack.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

dtypes.py

Copybara import of the project:

2023-07-24 14:38:20 -07:00

errors.py

Tracebacks no longer have JAX-internal frames prepended by default

2023-08-03 11:38:38 -07:00

flatten_util.py

ravel_pytree now accepted by static linters

2023-03-07 23:35:48 +00:00

linear_util.py

JEX: move jax.linear_util to jax.extend.linear_util

2023-08-30 18:32:12 -07:00

monitoring.py

Allow duration event listeners to take extra keyword arguments.

2023-10-04 13:15:49 -07:00

prng.py

Deprecate the contents of jax.prng

2023-08-30 15:13:32 -07:00

profiler.py

Document why 'import name as name' is used

2022-12-14 15:07:04 -08:00

py.typed

Improve type annotations

2021-01-13 10:26:35 -05:00

random.py

[random] deprecate jax.random.default_prng_impl()

2023-10-19 13:59:01 -07:00

sharding.py

Delete OpShardingSharding export since it has been 3 months since it was deprecated. Also remove deprecation warnings for MeshPspecSharding.

2023-06-08 13:53:38 -07:00

stages.py

add CompilerOptions type alias for AOT compiler options

2023-04-12 16:49:13 -07:00

test_util.py

Document why 'import name as name' is used

2022-12-14 15:07:04 -08:00

tree_util.py

fix typo in deprecation message

2023-10-03 14:04:09 -07:00

typing.py

Define jax.typing.DTypeLike

2023-10-10 08:46:36 -07:00

util.py

Document why 'import name as name' is used

2022-12-14 15:07:04 -08:00

version.py

Finish 0.4.19 jax and jaxlib release

2023-10-19 13:27:52 -07:00