Sergei Lebedev 5cad02c1f9 [pjrt] Use the PjRtMemorySpace version of BufferFromHostLiteral ...

Buffers live in memory spaces and not on devices. The `PjRtDevice` version
of `BufferFromHostLiteral` is deprecated and will be removed once the migration
is complete.

PiperOrigin-RevId: 721860983

2025-01-31 12:22:23 -08:00

..

BUILD

Move JAX example to public XLA:CPU API

2024-11-19 14:19:29 -08:00

main.cc

[pjrt] Use the PjRtMemorySpace version of BufferFromHostLiteral

2025-01-31 12:22:23 -08:00

prog.py

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

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