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