class_docstr: Exponentially decaying scheduler.
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."""
      if step >= self.num_steps:
          return jnp.full_like(
              step, self.final_value, dtype=jnp.float32
          )
  
      final_value = max(self.final_value, self.eps)
      base = final_value / self.initial_value
      exponent = step / (self.num_steps - 1)
      if step >= self.num_steps:
          return jnp.full_like(
              step, self.final_value, dtype=jnp.float32
          )
      return self.initial_value * base**exponent
  
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 ExponentialSchedule obj's underlying member method code with example usage: obj.get(step) and returns: unknown, jnp, jnp