Sharded data on a single host#

Note

This is a placeholder for a section in the new JAX tutorials.