spdl.io.to_jax

to_jax(buffer: CPUBuffer)

Convert to JAX Array.

Parameters:

buffer – Object implements the array interface protocol.

Returns:

A JAX Array.

Return type:

(jax.Array)