dogear

enter for all results · esc to close

Compiling machine learning programs via high-level tracing. Roy Frostig, Matthew James Johnson, Chris Leary. MLSys 2018.

mlsys.orgsite

White paper describing an early version of JAX, detailing how computation is traced and compiled.

visit mlsys.org ↗
0
sign in to dogear this
from
JAX
added
2026-10-10
likes
0

similar

  1. JAX: Accelerated Machine Learning Research | SciPy 2020 | VanderPlas youtube.com

    JAX's core design, how it's powering new research, and how you can start using it.

    video · 24 min · Enthought topic aitopic design link ↗

  2. Getting started with JAX (MLPs, CNNs & RNNs) by Robert Lange roberttlange.github.io

    Neural network building blocks from scratch with the basic JAX operators.

    site topic ai link ↗

  3. Jax in Action manning.com

    A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.

    site topic ai link ↗

  4. SymJAX github.com

    Symbolic CPU/GPU/TPU programming.

    tool topic programming link ↗

  5. JAX: Accelerated machine-learning research via composable function transformations in Python | NeurIPS 2019 | Skye… slideslive.com

    JAX intro presentation in Program Transformations for Machine Learning workshop.

    site topic aitopic programming link ↗

  6. Accurate Quantized Training github.com

    Tools and libraries for running and analyzing neural network quantization experiments in JAX and Flax.

    site topic ai link ↗

JAX › Papers: “White paper describing an early version of JAX, detailing how computation is traced and compiled.”