- Adding support for the "Schedule-free" method to be used as a wrapper for Optax optimizers in the examples codebase.

- Adding support for the "Schedule-free" method to be used as a wrapper for Optax optimizers in the examples codebase. #1359