Peter Hawkins e02faabfb2 Replace references to jax.readthedocs.io with docs.jax.dev. ...

PiperOrigin-RevId: 745156931

2025-04-08 08:33:49 -07:00

..

core

make random_gamma_grad not a primitive anymore

2025-04-01 17:04:50 +00:00

mlir

Remove the try/except for Shardy imports.

2025-04-01 11:33:44 -07:00

__init__.py

Replace references to jax.readthedocs.io with docs.jax.dev.

2025-04-08 08:33:49 -07:00

backend.py

Use JAX's default device instead of jax.devices()[0], if set.

2024-12-03 16:53:58 -08:00

BUILD

Export jax.lib.xla_extension.ifrt_programs.

2024-09-18 09:28:56 -07:00

ffi.py

Move jex.ffi to jax.ffi.

2024-12-29 13:06:19 +00:00

ifrt_programs.py

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

linear_util.py

[better_errors] Make it explicit that debug_info is not None.

2025-02-13 22:07:04 -08:00

random.py

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

source_info_util.py

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00