class_docstr: Ensures Sonnet optimizers have equivalent behavior to TensorFlow.
class_docstr_tok: ["Ensures","Sonnet","optimizers","have","equivalent","behavior","to","TensorFlow","."]
code: def _make_tf(self, learning_rate, beta_1, beta_2, epsilon):
      optimizer = tf.optimizers.Adam(
          learning_rate=learning_rate,
          beta_1=beta_1,
          beta_2=beta_2,
          epsilon=epsilon,
      )
      return lambda g, p: optimizer.apply_gradients(zip(g, p))
  
file_path: sonnet/sonnet/src/optimizers/adam_test.py
imports: from sonnet.src import test_utils
  from sonnet.src.optimizers import adam
  from sonnet.src.optimizers import optimizer_tests
  import tensorflow
label_desc: Use image node_id 1 for calling the ComparisonTest obj's underlying member method code with example usage: obj._make_tf(learning_rate, beta_1, beta_2, epsilon) and returns: function