More Pyrefly fixes in Pallas

Commit

65 days ago

More Pyrefly fixes in Pallas The remaining errors require us to bundle more .pyis with jaxlib.