LilyEvansHogwarts / BNN

Bayesian neural network implementations

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

README

Implementation of several Bayeisan neural networks

Implemented:

  • MC dropout
  • SGLD
  • SVI (from pyro)
  • BBB

TODO

  • Variational continual learning (for online learning)
  • Neural process
  • Stein Variational Gradient Descent
  • Particle mirror descent
  • SGLD with (stochastic) MH testing

About

Bayesian neural network implementations


Languages

Language:Python 100.0%