src/transformers/models/reformer/modeling_reformer.py/PositionEmbeddings/__init__
class PositionEmbeddings:    def __init__(self, config):
        super().__init__()
        self.dropout = config.hidden_dropout_prob
        self.embedding = nn.Embedding(config.max_position_embeddings, config.hidden_size)