Dan Foreman-Mackey cb4d97aa1f Move jex.ffi to jax.ffi.

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

..

core

Introduce lax.ragged_all_to_all primitive

2024-12-09 22:19:40 -08: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

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

Stop using generators for linear_util transformations.

2024-11-13 13:47:07 -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