class_docstr: Optax-like optimizer for testing pure functional optimization in TF.
class_docstr_tok: ["Optax-like","optimizer","for","testing","pure","functional","optimization","in","TF","."]
code: def __init__(self, learning_rate, decay_learning_rate=False):
      self._learning_rate = learning_rate
      self._decay_learning_rate = decay_learning_rate
  
imports: import collections
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python import optimizer
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.math.minimize import minimize
  from tensorflow_probability.python.math.minimize import minimize_stateless
label_desc: Use image node_id 1 to create a new StatelessSGD object from inherited base classes: object with example: obj = StatelessSGD(learning_rate, decay_learning_rate)