class_docstr: This class implements a classifier with the Jax framework.
class_docstr_tok: ["This","class","implements","a","classifier","with","the","Jax","framework","."]
code: def compute_loss(
      self, x: np.ndarray, y: np.ndarray, **kwargs
  ) -> np.ndarray:
      raise NotImplementedError
  
imports: from __future__ import absolute_import, division, print_function, unicode_literals
  import logging
  import random
  from typing import Any, Callable, Dict, List, Optional, Tuple, Union, TYPE_CHECKING
  import numpy
  from art.estimators.classification.classifier import ClassGradientsMixin, ClassifierMixin
  from art.experimental.estimators.jax import JaxEstimator
label_desc: Use image node_id 16 for calling the JaxClassifier obj's underlying member method code with example usage: obj.compute_loss(x, y) without return types