def preconditioner_from_params(param):
        """Returns a Preconditioner object for given param."""
        return Preconditioner(
            param,
            block_size,
            merge_small_dims_block_size,
            best_effort_shape_interpretation,
            precondtioner_type,
        )
Returns a Preconditioner object for given param.
tools/train/scalable_shampoo/distributed_shampoo.py