jax.experimental.pallas.empty_like

jax.experimental.pallas.empty_like(x)

Create an empty PyTree of possibly uninitialized values.

Parameters:

x (object) – A PyTree with leaves specifying the shape and dtype of the uninitialized object.

Returns:

A PyTree with the same structure as x, but with uninitialized values.

See also

jax.lax.empty()