class_docstr: Schedule that eases slowsly using a cosine.
docstr: Get the value for the given step.
docstr_tok: ["Get","the","value","for","the","given","step","."]
code: def get(self, step):
      """Get the value for the given step."""
      alpha = jnp.minimum(step / self.num_steps, 1.0)
      scale = self.final_value - self.initial_value
      x = min(max(alpha, 0.0), 1.0)
      return self.initial_value + scale * 0.5 * (
          1 + math.cos(jnp.pi * x + jnp.pi)
      )
  
imports: import abc
  import collections
  import copy
  import math
  from typing import Any, Iterable, List, Tuple, Union
  from jax import numpy
label_desc: Use image node_id 2 for calling the CosineEasingSchedule obj's underlying member method code with example usage: obj.get(step) and returns: unknown