← all repositories

piEsposito/blitz-bayesian-deep-learning

A PyTorch library providing Bayesian neural network layers with weight uncertainty for regression and classification.

981 stars Python ML Frameworks
blitz-bayesian-deep-learning
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

BLiTZ is a Python library that extends PyTorch with Bayesian neural network layers implementing weight uncertainty as described in the seminal paper on Weight Uncertainty in Neural Networks. It provides variational layers where weights are sampled from learned distributions rather than fixed values, enabling uncertainty quantification in deep learning models. The library integrates seamlessly with standard PyTorch workflows, allowing users to add probabilistic modeling capabilities to existing architectures while maintaining compatibility with standard layers and training loops.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.