realised pytensor might be relevant too - I’ve got 2.23.0. Tried upgrading to 2.24.1 but it’s incompatible with pymc 5.16.1