47264877a1
a_min and a_max are deprecated parameter names to jax.numpy.clip. PiperOrigin-RevId: 891786675 Change-Id: Icc06efe0381c0db06f4aeb16f281a4d01ea37539