Implement Differentiable for another package of distributions