I work on the foundations of machine learning, by way of theory as well as systems and programming languages. Percy Liang was my PhD advisor at Stanford, where I was part of the statistical machine learning group.
I created JAX with a few colleagues in 2017. We're still working on it.
I contributed a bit to the Rust programming language in its early stages.
Publications, preprints, and abstracts [also on scholar]: