code: def adam_update_numpy(
      param,
      g_t,
      t,
      m,
      v,
      lr=0.001,
      beta1=0.9,
      beta2=0.999,
      epsilon=1e-7,
  ):
      lr_t = lr * np.sqrt(1 - beta2 ** (t + 1)) / (1 - beta1 ** (t + 1))
  
      m_t = beta1 * m + (1 - beta1) * g_t
      v_t = beta2 * v + (1 - beta2) * g_t * g_t
  
      param_t = param - lr_t * m_t / (np.sqrt(v_t) + epsilon)
      return param_t, m_t, v_t
  
file_path: addons/tensorflow_addons/optimizers/tests/lazy_adam_test.py
imports: import numpy
  import tensorflow
  from tensorflow_addons.optimizers import lazy_adam
  from tensorflow_addons.utils import test_utils
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: adam_update_numpy(param, g_t, t, m, v, lr, beta1, beta2, epsilon) and returns: param_t,  m_t,  v_t