Roy Frostig 9e7113e6ed add extensibility pointers to jax.extend docstring

2024-11-07 07:32:20 -08:00

..

core

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

mlir

Drop support for mhlo in JAX's public API.

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

__init__.py

add extensibility pointers to jax.extend docstring

2024-11-07 07:32:20 -08:00

backend.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

BUILD

Export jax.lib.xla_extension.ifrt_programs.

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

ffi.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

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

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

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