flax

c9965eb4 - Adds startup script launch_gce.

Adds startup script launch_gce. Verified for [mixed precision] 8x V100 imagenet: https://tensorboard.dev/experiment/rbQrlIgrTwaV16ka0ZawFw/