class_docstr: Test optimizers and related classes.
class_docstr_tok: ["Test","optimizers","and","related","classes","."]
docstr: Test creating an optimizer with an exponentially decaying learning rate.
docstr_tok: ["Test","creating","an","optimizer","with","an","exponentially","decaying","learning","rate","."]
code: def test_exponential_decay_jax(self):
      """Test creating an optimizer with an exponentially decaying learning rate."""
      rate = optimizers.ExponentialDecay(
          initial_rate=0.001, decay_rate=0.99, decay_steps=10000
      )
      opt = optimizers.Adam(learning_rate=rate)
      jaxopt = opt._create_jax_optimizer()
      assert isinstance(jaxopt, optax.GradientTransformation)
  
imports: import deepchem
  import deepchem.models.optimizers
  import unittest
  import pytest
label_desc: Use image node_id 20 for calling the TestOptimizers obj's underlying member method code with example usage: obj.test_exponential_decay_jax() without return types