For the time being, I have restricted numpy to <=1.26.4 in requirements. As numpy 2.X matures, and tensorflow catches up, we should revisit to lift this restriction asap.