Independent project. Not a U.S. government website.

Support our project (opens Buy Me a Coffee)
USASI

Jump to

Start hereFour ways into the catalogCompanies & LabsU.S. AI organizationsOpen Models & ToolsOpen models, software, datasets, and evaluation toolsLearnExplainers and learning pathsFor kidsAI and Super Intelligence, explained for readers 8 to 12Latest newsSourced news itemsPolicy trackerFederal AI laws, orders, and standards

Jake VanderPlas

Named author of JAX who has written several of its design proposals

  • Company listed on Jake VanderPlas's GitHub profile, Google1

Jake VanderPlas is one of the authors named in the JAX repository's citation entry. The JAX documentation lists VanderPlas as the author of design proposals on JAX's type promotion semantics (December 2021), the scope of its NumPy and SciPy wrappers (October 2023), and stateful randomness (November 2025), and as a co-author, with Roy Frostig, of the proposal for typed PRNG keys. VanderPlas's GitHub profile lists Google as their company.245671

Work

  • JAXOpen-source code

    Named in the "Citing JAX" author list; author of JAX design documents on type promotion, the scope of jax.numpy and jax.scipy (JEP 18137), and stateful randomness (JEP 28845), and co-author of JEP 9263 on typed keys. The repository is released under the Apache License 2.0.245673

    Project page (external site: github.com)

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.

Sources · reviewed Oct 13, 2026

  1. 1.
    jakevdp (Jake VanderPlas) on GitHub (external site: github.com)

    GitHub · Official page · accessed Oct 13, 2026

  2. 2.
    jax-ml/jax README ("Citing JAX" author list) (external site: github.com)

    jax-ml (GitHub repository) · Repository · accessed Oct 13, 2026

  3. 3.
    jax-ml/jax LICENSE (Apache License 2.0) (external site: github.com)

    jax-ml (GitHub repository) · License · accessed Oct 13, 2026

  4. 4.
    Design of Type Promotion Semantics for JAX (external site: docs.jax.dev)

    JAX documentation · Documentation · published Dec 2021 · accessed Oct 13, 2026

  5. 5.
    JEP 18137: Scope of JAX NumPy & SciPy Wrappers (external site: docs.jax.dev)

    JAX documentation · Documentation · published Oct 2023 · accessed Oct 13, 2026

  6. 6.
    JEP 28845: Stateful Randomness in JAX (external site: docs.jax.dev)

    JAX documentation · Documentation · published Nov 2025 · accessed Oct 13, 2026

  7. 7.
    JEP 9263: Typed keys & pluggable RNGs (external site: docs.jax.dev)

    JAX documentation · Documentation · published Aug 2023 · accessed Oct 13, 2026

Support Us

Help keep USASI useful.

Leave a tip on Buy Me a Coffee (external site)

Optional. No USASI account required. Payment takes place on the linked provider’s website (Buy Me a Coffee).

About supporting this project