jax

0adb584b - Marked the parameters of safe_zip, safe_map and foreach as positional-only

Marked the parameters of safe_zip, safe_map and foreach as positional-only This matches the signature of the original builtins as well as the impl in jaxlib/utils.cc PiperOrigin-RevId: 836249444