Peter Hawkins 1baed9b285 [PJRT:CPU] Replace references to pjrt/tfrt_cpu_pjrt_client with pjrt/cpu/cpu_client.h. ...

The two are aliases and the former is a forwarding header pointing to the latter.

Cleanup only, no functional changes.

PiperOrigin-RevId: 621341188

2024-04-02 17:20:16 -07:00

..

jax_cpp

[PJRT:CPU] Replace references to pjrt/tfrt_cpu_pjrt_client with pjrt/cpu/cpu_client.h.

2024-04-02 17:20:16 -07:00

__init__.py

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

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

advi.py

update top-level examples to use new-style typed keys

2024-03-07 12:40:10 -08:00

datasets.py

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

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

differentially_private_sgd.py

update top-level examples to use new-style typed keys

2024-03-07 12:40:10 -08:00

examples_test.py

update top-level examples to use new-style typed keys

2024-03-07 12:40:10 -08:00

gaussian_process_regression.py

update top-level examples to use new-style typed keys

2024-03-07 12:40:10 -08:00

kernel_lsq.py

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

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

mnist_classifier_fromscratch.py

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

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

mnist_classifier.py

update top-level examples to use new-style typed keys

2024-03-07 12:40:10 -08:00

mnist_vae.py

update top-level examples to use new-style typed keys

2024-03-07 12:40:10 -08:00

onnx2xla.py

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

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

spmd_mnist_classifier_fromscratch.py

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

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