Hamiltonain Monte Carlo in Python