Peter Hawkins
172a831219
Switch JAX to use the OpenXLA repository.
2023-03-13 18:38:26 +00:00
Peter Hawkins
88379603e0
[PJRT] Delete the old :cpu_device target that uses StreamExecutor.
...
The TFRT CPU client is better in every way and the SE CPU client is unmaintained and has not been used by JAX in many months.
PiperOrigin-RevId: 489246256
2022-11-17 10:29:03 -08:00
Peter Hawkins
ba557d5e1b
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
...
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
2022-09-22 12:27:19 -07:00
Jean-Baptiste Lespiau
2929d79d0b
Use value instead of ValueOrDie.
...
PiperOrigin-RevId: 473011038
2022-09-08 09:35:52 -07:00
jax authors
32e77772b9
Rename xla::PjRtExecutable to xla::PjRtLoadedExecutable
...
PiperOrigin-RevId: 463460929
2022-07-26 17:06:45 -07:00
jax authors
d4c6f95331
Rename sync versions of ToLiteral to ToLiteralSync to facilitate upcoming refactor.
...
PiperOrigin-RevId: 432285004
2022-03-03 14:29:51 -08:00
Qiao Zhang
7e155244b6
Add example code to save JAX program and run using C++ runtime.
2021-01-25 19:12:26 -08:00
