warp.to_jax

warp.to_jax(warp_array)

Convert a Warp array to a Jax array without copying the data.

Parameters:

warp_array (warp.array) – The Warp array to convert.

Returns:

The converted Jax array.

Return type:

jax.Array