Roy Frostig
Co-author and core developer of JAX; research scientist at Google DeepMind
- Research scientist, Google DeepMind12
Roy Frostig is a research scientist at Google DeepMind and a co-author and core developer of JAX, which Frostig's homepage says Frostig created with a few colleagues in 2017. With Matthew Johnson and Chris Leary, Frostig wrote the SysML 2018 paper "Compiling machine learning programs via high-level tracing," which reported on an early version of JAX. Frostig's homepage describes work on the foundations of machine learning through theory, systems, and programming languages, and the JAX documentation lists Frostig as an author of design proposals including typed PRNG keys and the jax.extend module.13467
Work
JAXOpen-source code
Co-author and core developer of JAX according to Frostig's homepage; named in the "Citing JAX" author list; co-author of the SysML 2018 paper on an early JAX; author or co-author of JAX design proposals on typed keys and pluggable RNGs (August 2023) and on jax.extend. The repository is released under the Apache License 2.0.143675
This profile covers public professional work only, from the sources below. It is not an endorsement or a ranking. To correct or remove it, use the correction form.