search

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

Add mixture distribution for pytorch #40

Closed brendanhasz closed 3 years ago

brendanhasz commented 3 years ago

They've implemented a MixtureSameFamily distribution for PyTorch, so add the pytorch implementation for probflow.distributions.Mixture using that.

Fixes https://github.com/brendanhasz/probflow/issues/9