docstr: Test rastrigin loss for Adam optimizer.
docstr_tok: ["Test","rastrigin","loss","for","Adam","optimizer","."]
code: def test_rastrigin_adam(self):
      """Test rastrigin loss for Adam optimizer."""
      model_vars, loss = self._test_loss_opt("rastrigin", "adam")
  
imports: import math
  import unittest
  import jax.numpy
  import numpy
  import objax
  from objax.variable import TrainVar, VarCollection
label_desc: Use image node_id 7 for calling the TestOptimizers obj's underlying member method code with example usage: obj.test_rastrigin_adam() without return types