AJAX
This repo contains the ADVI implementation in JAX. The original paper is here.
Installation
pip install git+https://github.com/patel-zeel/ajax.git
Basic usage
from ajax import Prior, Likelihood, Variational, ADVI
This repo contains the ADVI implementation in JAX. The original paper is here.
pip install git+https://github.com/patel-zeel/ajax.git
from ajax import Prior, Likelihood, Variational, ADVI