examples/research_projects/performer/modeling_flax_performer_utils.py/GaussianUnstructuredRandomMatrix/get_2d_array
class GaussianUnstructuredRandomMatrix:    def get_2d_array(self):
        return random.normal(self.key, (self.nb_rows, self.nb_columns))