text: from . import experimental, nn
  from .filters import (
      combine,
      filter,
      is_array,
      is_array_like,
      is_inexact_array,
      is_inexact_array_like,
      partition,
  )
  from .grad import filter_custom_vjp, filter_grad, filter_value_and_grad
  from .jit import filter_jit
  from .module import Module, static_field
  from .tree import tree_at, tree_equal, tree_pformat
  from .update import apply_updates
  
  
  __version__ = "0.3.1"
  