Thanks for trying to reproduce.
Yes, I am working with the latest commit from the PyMC3 main branch.
main