Hi, that environment was running pymc 5.10.0. I created a fresh environment with 5.10.3, and I no longer have the bugs. No worries about the divergences, I just created the example quickly to recreate the bug.
Unclear if the original environment was somehow corrupted or if it was just the update that fixed the problem.
Should have just tried that in the in the first place, thanks a lot for the help and sorry for the waste of time.