Currently working on JAX (https://github.com/google/jax) at Google Brain Former: tech lead & manager for the Swift for TensorFlow (S4TF) team in Google Brain