-record(linear_schedule, {
    beta_start :: float(),
    beta_end :: float(),
    num_steps :: integer()
}).