├── README.md ├── examples ├── Logistic regression example.ipynb ├── Tennis example.ipynb ├── sackmann.py ├── stan_log_reg.stan ├── stan_tennis.stan └── stan_utils.py ├── images ├── bradley_terry.png ├── log_reg_model.png ├── logistic_reg_comparison.png ├── tennis_comparison.png └── tennis_comparison_stan_advi.png ├── jax_advi ├── __init__.py ├── advi.py ├── constraints.py ├── lrvb.py ├── optimization.py └── utils │ ├── __init__.py │ ├── autodiff.py │ ├── flattening.py │ └── misc.py └── setup.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/README.md -------------------------------------------------------------------------------- /examples/Logistic regression example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/examples/Logistic regression example.ipynb -------------------------------------------------------------------------------- /examples/Tennis example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/examples/Tennis example.ipynb -------------------------------------------------------------------------------- /examples/sackmann.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/examples/sackmann.py -------------------------------------------------------------------------------- /examples/stan_log_reg.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/examples/stan_log_reg.stan -------------------------------------------------------------------------------- /examples/stan_tennis.stan: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/examples/stan_tennis.stan -------------------------------------------------------------------------------- /examples/stan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/examples/stan_utils.py -------------------------------------------------------------------------------- /images/bradley_terry.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/images/bradley_terry.png -------------------------------------------------------------------------------- /images/log_reg_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/images/log_reg_model.png -------------------------------------------------------------------------------- /images/logistic_reg_comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/images/logistic_reg_comparison.png -------------------------------------------------------------------------------- /images/tennis_comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/images/tennis_comparison.png -------------------------------------------------------------------------------- /images/tennis_comparison_stan_advi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/images/tennis_comparison_stan_advi.png -------------------------------------------------------------------------------- /jax_advi/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /jax_advi/advi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/jax_advi/advi.py -------------------------------------------------------------------------------- /jax_advi/constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/jax_advi/constraints.py -------------------------------------------------------------------------------- /jax_advi/lrvb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/jax_advi/lrvb.py -------------------------------------------------------------------------------- /jax_advi/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/jax_advi/optimization.py -------------------------------------------------------------------------------- /jax_advi/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /jax_advi/utils/autodiff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/jax_advi/utils/autodiff.py -------------------------------------------------------------------------------- /jax_advi/utils/flattening.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/jax_advi/utils/flattening.py -------------------------------------------------------------------------------- /jax_advi/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/jax_advi/utils/misc.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martiningram/jax_advi/HEAD/setup.py --------------------------------------------------------------------------------