jax

9504a479 - [Mosaic GPU][NFC] Rename `EquationSystem` to `ConstraintSystem`.

[Mosaic GPU][NFC] Rename `EquationSystem` to `ConstraintSystem`. I decided to alias the constraints module as `cs` because `constraints` is widely used as a variable name and it resulted in a lot of conflicts. PiperOrigin-RevId: 836308644