Remove deprecated jax.experimental.shard_map import from jax.experimental.checkify.
PiperOrigin-RevId: 972126306
P
Peter Hawkins committed
70dde6661fbf1992b161e1511fb04d8c8665cc26
Parent: c33b927
Committed by jax authors <google-ml-automation@google.com>
on 8/27/2026, 8:32:57 PM