Differentiable Random Partition Models