examples/research_projects/performer/modeling_flax_performer_utils.py/FastAttentionviaLowRankDecomposition/draw_weights
class FastAttentionviaLowRankDecomposition:    def draw_weights(self, key):
        if self.matrix_creator is None:
            return None
        matrixrng, _ = random.split(key)
        projection_matrix = self.matrix_creator(key=matrixrng).get_2d_array()
        return projection_matrix