src/transformers/models/jukebox/modeling_jukebox.py/JukeboxRangeEmbedding/__init__
class JukeboxRangeEmbedding:    def __init__(self, n_time, embed_dim, range, out_width, clamp=False):
        super().__init__()
        self.n_time = n_time
        self.embed_dim = embed_dim
        self.emb = nn.Embedding(embed_dim, out_width)
        self.pos_min, self.pos_max = range
        self.clamp = clamp