class_docstr: A type alias for apply functions.
  
  Apply functions do computations with finite-width neural networks. They are
  functions that take a PyTree of parameters and an array of inputs and produce
  an array of outputs.
class_docstr_tok: ["A","type","alias","for","apply","functions",".","Apply","functions","do","computations","with","finite-width","neural","networks",".","They","are","functions","that","take","a","PyTree","of","parameters","and","an","array","of","inputs","and","produce","an","array","of","outputs","."]
code: def __call__(
      self, params: PyTree, inputs: NTTree[jnp.ndarray], *args, **kwargs
  ) -> NTTree[jnp.ndarray]:
      ...
  
imports: from typing import Any, Generator, Optional, Protocol, Sequence, TYPE_CHECKING, TypeVar, Union
  import jax
  import jax.numpy
  from .kernel import Kernel
label_desc: Use image node_id 1 for calling the ApplyFn obj's underlying member method code with example usage: obj.__call__(params, inputs) without return types