贝叶斯深度学习的 GPU 库 ZhuSuan

• 授权协议: MIT
• 开发语言: Python
• 操作系统: 跨平台
• 软件首页: https://github.com/thu-ml/zhusuan
• 软件文档: http://zhusuan.readthedocs.io/en/latest/
• 官方下载: https://github.com/thu-ml/zhusuan

软件介绍

ZhuSuan （珠算）是清华大学机器学习组开源的一个贝叶斯深度学习的 GPU 库，是一个构建于 TensorFlow 之上的用于生成模型的 Python 库，结合了贝叶斯方法和深度学习的优势进行互补。与目前主要面向神经网络和监督任务的深度学习库不同，珠算的特点是深入到贝叶斯推断中，从而支持各类概率模型，包括传统分层贝叶斯模型以及最近的深度生成模型。

支持的推理

• 变分推理和随机变分推理 (SGVB、REINFORCE、VIMCO 等)
  

• 自适应重要程度采样
  

• 马尔可夫链蒙特卡洛（MCMC）

示例

项目提供传统分层贝叶斯模型和近期的深度生成模型的实例。要运行提供的示例，你可能需要安装额外的依赖包。可以通过执行以下代码完成：

pip install ".[examples]"

• Gaussian: HMC

• Toy 2D Intractable Posterior: SGVB

• Bayesian Neural Networks: SGVB

• Variational Autoencoder (VAE): SGVB, IWAE

• Convolutional VAE: SGVB

• Semi-supervised VAE (Kingma, 2014): SGVB, RWS

• Deep Sigmoid Belief Networks RWS, VIMCO

• Logistic Normal Topic Model: HMC

本文地址：https://codercto.com/soft/d/16748.html