examples/research_projects/performer/modeling_flax_performer_utils.py/GaussianOrthogonalRandomMatrix/__init__
class GaussianOrthogonalRandomMatrix:    def __init__(self, nb_rows, nb_columns, key, scaling=0):
        self.nb_rows = nb_rows
        self.nb_columns = nb_columns
        self.key = key
        self.scaling = scaling