jax

d33a5689 - Refactor & test internal deprecation APIs

Refactor & test internal deprecation APIs The names and APIs were previously too similar and therefore somewhat confusing; this will be more clear I think. PiperOrigin-RevId: 635615163