You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.
Stochastic HybridBlock and HybridSequential: they build upon gluon HybridBlock and HybridSequential and allows adding extra loss to each layer.
Distribution class: it implements a variety of functionalities including prob, log_prob, sample, broadcast_to, mean, variance, etc.
KL divergence: kl_divergence(p, q) function searches over registered KL divergence functions and performs computation.
Transform: transform one distribution to another invertible distribution.
Independent: reinterprets some of the batch dims of a distribution as event dims.
Two features that is currently either not supported or kind of broken in MxNet will be very useful to this projects: symbolic shape and control flow.
At the moment, we will implement most of distribution in frontend. We will move the computation to backend when new numpy probability ops such as chisquare, dirichlet and multivariate_normal are introduced into MxNet.
Description
In MxNet 2.0, we would like to provide a distribution module, analogous to Pytorch distribution. The main difference from theirs is that we use numpy op and it allows hybridization. The current project code can be seen from https://github.com/xidulu/incubator-mxnet/tree/distribution_dev/python/mxnet/gluon/probability.
The basic skeleton divides into following parts:
HybridBlock
andHybridSequential
: they build upon gluonHybridBlock
andHybridSequential
and allows adding extra loss to each layer.prob
,log_prob
,sample
,broadcast_to
,mean
,variance
, etc.kl_divergence(p, q)
function searches over registered KL divergence functions and performs computation.Two features that is currently either not supported or kind of broken in MxNet will be very useful to this projects: symbolic shape and control flow.
At the moment, we will implement most of distribution in frontend. We will move the computation to backend when new numpy probability ops such as
chisquare
,dirichlet
andmultivariate_normal
are introduced into MxNet.References
@xidulu @szha @leezu @haojin2
The text was updated successfully, but these errors were encountered: