James Bradbury
Named author of JAX and co-author of the original PyTorch paper
James Bradbury is one of the authors named in the JAX repository's citation entry and a co-author of the 2019 PyTorch paper, where Bradbury is listed with Google. The JAX documentation also credits Bradbury, under the handle jekbradbury, as a co-author of the January 2023 design doc that proposed shard_map for per-device code.14536
Work
JAXOpen-source code
Named in the "Citing JAX" author list of the JAX README and credited as a co-author of the shard_map design doc (January 2023) in the JAX documentation. The repository is released under the Apache License 2.0.132
PyTorchPublished research
Co-author of "PyTorch: An Imperative Style, High-Performance Deep Learning Library" (2019).4
Other documented roles
Roles the sources document but that are not confirmed as current.
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.