Is there something like the DistributionLambda layer from Keras in Flux?

DistributionLambda

I guess it could use Distributions.jl.