dogear

enter for all results · esc to close

Deterministic ADVI in JAX by Martin Ingram

martiningram.github.iosite

Walk through of implementing automatic differentiation variational inference (ADVI) easily and cleanly with JAX.

visit martiningram.github.io ↗
0
sign in to dogear this
from
JAX
added
2026-10-10
likes
0

similar

  1. Jax in Action manning.com

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

    site topic ai link ↗

  2. GPJax github.com

    Gaussian processes in JAX.

    tool link ↗

  3. Introduction to JAX youtube.com

    Simple neural network from scratch in JAX.

    video · 7 min · Google Cloud Tech topic ai link ↗

  4. Enabling Fast Differentially Private SGD via Just-in-Time Compilation and Vectorization. Pranav Subramani, Nicholas… arxiv.org

    Uses JAX's JIT and VMAP to achieve faster differentially private than existing libraries.

    site link ↗

  5. Evolving Neural Networks in JAX by Robert Tjarko Lange roberttlange.github.io

    Explores how JAX can power the next generation of scalable neuroevolution algorithms.

    site topic aitopic programming link ↗

  6. kalman-jax github.com

    Approximate inference for Markov (i.e., temporal) Gaussian processes using iterated Kalman filtering and smoothing.

    tool link ↗

JAX › Tutorials and Blog Posts: “Walk through of implementing automatic differentiation variational inference (ADVI) easily and cleanly with JAX.”