nip.utils.distributions# PyTorch distributions. Classes CompositeCategoricalDistribution(key_transform) A composition of categorical distributions.