I am trying to repeat Bayesian neural network (i.e: a simple binary classification problem that’s not linearly separable) from the PyMC3 tutorial case “Variational Inference: Bayesian Neural Networks”. However, it is much slower than that showed in the case (speed up to 3139.84it/s). The code and data used were the same with the case.
Actual result: