jax.experimental.pallas.max_contiguous#

jax.experimental.pallas.max_contiguous(x, values)[source]#

A compiler hint that asserts the values first values of x are contiguous.