ADVI in PyMC3 3.2 is approximately two times slower than 3.1

what do you mean by step function?