benchmark

validating JAX acceleration