Not sure this repo could help you or not.
  
      

      GitHub
  

  
    

GitHub - piEsposito/blitz-bayesian-deep-learning: A simple and extensible... 29

  A simple and extensible library to create Bayesian Neural Network layers on PyTorch. - GitHub - piEsposito/blitz-bayesian-deep-learning: A simple and extensible library to create Bayesian Neural Ne...