A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch
BSD 3-Clause "New" or "Revised" License
544
stars
73
forks
source link
Added Skew geometric jensen-shannon divergence #35
Open
rewat7 opened 9 months ago
Added function for calculating skew-geometric jensen shannon divergence instead of kl-divergence with minimal code changes