NUTS and zero derivative error

You shouldnt assign step method yourself unless you have special use case - try just trace = pm.sample()