class_docstr: This class implements a classifier with the Jax framework.
class_docstr_tok: ["This","class","implements","a","classifier","with","the","Jax","framework","."]
docstr: Return the loss function.
  
  :return: The loss function.
docstr_tok: ["Return","the","loss","function",".",":","return",":","The","loss","function","."]
code: def loss_func(self) -> Callable:
      """
      Return the loss function.
  
      :return: The loss function.
      """
      return self._loss_func
  
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 5 for calling the JaxClassifier obj's underlying member method code with example usage: obj.loss_func() and returns: self