jax

70d266ee - [Mosaic GPU] Fix error message.

[Mosaic GPU] Fix error message. `op` does not exist in this scope. Funnily enough, python still prints a random op if that message is printed. PiperOrigin-RevId: 819093319