Kyle Lucke 84d748f43c Stop using xla/statusor.h now that it just contains an alias for absl::Status. ...

In some situations, this meant also changing unrelated files to directly include tsl/platform/statusor.h to get the definitions for TF_ASSIGN_OR_RETURN, etc., where they were getting transitively included for free.

PiperOrigin-RevId: 645169743

2024-06-20 15:09:40 -07:00

..

jax_cpp

Stop using xla/statusor.h now that it just contains an alias for absl::Status.

2024-06-20 15:09:40 -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

Force float32 matmuls in examples_test.

2024-05-10 19:30:02 +00:00

gaussian_process_regression.py

Avoid 'from jax import config' imports

2024-04-11 13:23:27 -07: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