JAX: commitment issues
A JAX developer found that array lookups on CPU-hosted data took over a second each, because arrays created with the `default_device` context manager are uncommitted to that device. Using `jax.device_put` to explicitly commit the array to the CPU reduced lookup times from ~1.2s to under 0.0002s, fixing a severe bottleneck in LLM training data loading.