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