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)
Convert to JAX Array.
buffer – Object that implements the array interface protocol
__array_interface__). ((has)
A JAX Array.
(jax.Array)