docstr: Returns step decay learning rate multiplier.
docstr_tok: ["Returns","step","decay","learning","rate","multiplier","."]
code: def multiplier(self, step: float):
      """Returns step decay learning rate multiplier."""
      if isinstance(self.step_size, (tuple, list)):
          exponent = jn.sum(
              jn.greater_equal(step, jn.array(self.step_size))
          )
      else:
          exponent = step // self.step_size
      return self.gamma**exponent
  
file_path: objax/objax/optimizer/scheduler.py
imports: import abc
  from typing import List, Tuple, Union
  import jax.numpy
label_desc: Use image node_id 2 for calling the StepDecay obj's underlying member method code with example usage: obj.multiplier(step) and returns: unknown