Working on scaling and serving large models
on the JAX team at Google Deepmind. resume