Add flipout pytorch - Githubissues

brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch

http://probflow.readthedocs.io

MIT License

171 stars 17 forks source link

Closed brendanhasz closed 3 years ago

brendanhasz commented 3 years ago

Add support for flipout with the PyTorch backend.
Add randn, rand_rademacher, and shape backend-independent ops
Update deprecated tfp.python.math.random_rademacher in favor of tfp.random.rademacher when possible