jax
4 posts
-
Building Non-Standard Regression Models with JAX
Using JAX's automatic differentiation to train a Zero-Inflated Generalised Poisson regression model
- jax
- regression
- count-data
- ZIP
- poisson
- maximum-likelihood
-
Online Bayesian Learning Part 3: R-VGA
Deriving and implementing the R-VGA algorithm for online learning with neural networks on streaming binary classification data
- bayesian
- neural-networks
- variational-inference
- jax
- equinox
- online
- r-vga
-
Online Bayesian Learning Part 2: Bayesian Neural Networks With Variational Inference
A quick detour to introduce variational inference and use it to train a small Bayesian neural network with jax/eqx
- bayesian
- neural-networks
- variational-inference
- jax
- equinox
-
Online Bayesian Learning Part 1: Exact recursive inference
Part 1 of a series of posts looking into the problem of 'online' learning using Bayesian methods
- bayesian
- neural-networks
- variational-inference
- jax
- equinox
- online