class_docstr: Like `functools.partial`, but treats the wrapped function, and partially-applied
  value: Any = field(static=True)
  args and kwargs, as a PyTree.
  
  This is very much like `jax.tree_util.Partial`. The difference is that the JAX
  version requires that `func` be specifically a *function* -- and will silently
  misbehave if given any non-function callable, e.g. [`equinox.nn.MLP`][]. In contrast
  the Equinox version allows for arbitrary callables.
docstr: Call the wrapped `self.func`.
  
  **Arguments:**
  
  - `*args`: the arguments to apply. Passed after those arguments passed during
      `__init__`.
  - `**kwargs`: any keyword arguments to apply.
  
  **Returns:**
  
  The result of the wrapped function.
docstr_tok: ["Call","the","wrapped","`","self.func","`",".","*","*","Arguments",":","*","*","-","`","*","args","`",":","the","arguments","to","apply",".","Passed","after","those","arguments","passed","during","`","__init__","`",".","-","`","*","*","kwargs","`",":","any","keyword","arguments","to","apply",".","*","*","Returns",":","*","*","The","result","of","the","wrapped","function","."]
code: def __call__(self, *args, **kwargs):
      """Call the wrapped `self.func`.
  
      **Arguments:**
  
      - `*args`: the arguments to apply. Passed after those arguments passed during
          `__init__`.
      - `**kwargs`: any keyword arguments to apply.
  
      **Returns:**
  
      The result of the wrapped function.
      """
      return self.func(*self.args, *args, **kwargs, **self.keywords)
  
imports: import abc
  import dataclasses
  import functools
  import inspect
  import types
  import warnings
  import weakref
  from collections.abc import Callable
  from typing import Any, cast, Optional, Protocol, TYPE_CHECKING, TypeVar, Union
  from typing_extensions import dataclass_transform, ParamSpec
  import jax._src.traceback_util
  import jax.tree_util
  import numpy
  from jaxtyping import Array, Bool, PyTreeDef
  from ._better_abstract import ABCMeta, dataclass
  from ._caches import internal_lru_caches
  from ._doc_utils import doc_repr
  from ._pretty_print import tree_pformat
  from ._tree import tree_equal
label_desc: Use image node_id 2 for calling the Partial obj's underlying member method code with example usage: obj.__call__() and returns: self