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

PiperOrigin-RevId: 676843138

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

..

__init__.py

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

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

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

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

mlir.py

Don't export ir_attribute from interpreters.mlir.

2024-08-14 07:53:18 -07:00

partial_eval.py

Remove some untested dynamic shapes paths (prep work for stackless).

2024-09-19 12:59:48 -07:00

pxla.py

Make pxla.shard_arg batch calls to xc.copy_array_to_devices_with_sharding

2024-06-13 13:10:10 -07:00

xla.py

Deprecate jax.interpreters xb, xc, xe abbreviations.

2024-06-28 12:38:43 -07:00