Roy Frostig

Research scientist, Google Research, Brain team
Co-author and core developer, JAX

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]:

➥  Supplemented by our dataset of training measurements.
➥  Reports on a nascent version of JAX.
➥  Corresponds to the initial release of SEMPRE.