Schedules

jaxnasium.algorithms.core.Schedule

A simple wrapper around optax schedules to allow for per-agent schedules in multi-agent settings. Is either a constant value or a linear schedule from start to end over the course of training.

__call__(count)