class_docstr: Adaptive Computation Time Function to help we use nn.scan on ACT.
code: def stop_fn(self, inputs: Any) -> jnp.ndarray:
      # Returns True if all of halting probability >= 1-eps.
      _, halting_prob, _, _, *_ = inputs
      return jnp.all(halting_prob >= self.threshold)
  
imports: from typing import Any, Callable, Optional, Sequence, Tuple
  from absl import logging
  import flax.linen
  import jax
  import jax.numpy
  import ml_collections
  from scenic.model_lib.layers import nn_layers
label_desc: Use image node_id 3 for calling the ACTFunction obj's underlying member method code with example usage: obj.stop_fn(inputs) and returns: jnp