Add large-scale AutoSharding example

[jeffcarp]

E.g. Gemma 27B.

Potentially switch to this implementation:
https://github.com/jax-ml/bonsai/tree/main/bonsai/models/gemma3