spdl.io.to_jax

to_jax(buffer: ArrayInterface)[source]

Convert to JAX Array.

Parameters:
  • buffer – Object that implements the array interface protocol

  • __array_interface__). ((has)

Returns:

A JAX Array.

Return type:

(jax.Array)