Expose method grad_log_prob

Closes #40.