Jake VanderPlas 3fa86a9b32 remove jax.extend.backend.default_backend in favor of jax.backend ...

I added this two days ago before realizing there is already a canonical API
for this in the top-level namespace, so it should be safe to remove.

2024-08-02 07:07:29 -07:00

..

core

re-export RNG primitives from jax.extend.core.primitives

2024-07-17 12:24:20 -07:00

mlir

Drop support for mhlo in JAX's public API.

2024-07-30 05:29:52 -07:00

__init__.py

Move jax.ffi submodule to jax.extend.ffi

2024-05-31 12:34:59 -04:00

backend.py

remove jax.extend.backend.default_backend in favor of jax.backend

2024-08-02 07:07:29 -07:00

BUILD

jax.extend.backend: add semi-private backend utils from xla_bridge

2024-07-31 16:19:30 -07:00

ffi.py

Export jaxlib.xla_client.register_custom_call_target as jax.extend.ffi.register_ffi_target.

2024-07-19 08:12:25 -04:00

linear_util.py

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

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

random.py

[jex] replace extend.random.PRNGImpl with extend.random.define_prng_impl

2023-10-31 17:21:54 -07:00

source_info_util.py

export jax.extend.source_info_util.current

2023-10-13 12:31:11 -07:00