docs: def _renorm(args: Dict[str, Any]):
      """Renormalizes the state using the norm arg."""
      state = _state_shard(args)
      # If our gate is so bad that we have norm of zero, we have bigger problems.
      state /= np.sqrt(args['norm_squared'])