-record(learned_positional_encoding, {
    max_len :: integer(),
    embedding_dim :: integer(),
    weight :: viva_tensor@tensor:tensor()
}).