jax.checkpoint_policies.checkpoint_dots# checkpoint_policies.checkpoint_dots = DotsSaveable(only_if_no_batch_dims=False)#
