old_contents: from . import experimental, nn
  from .eval_shape import filter_eval_shape
  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 .pretty_print import tree_pformat
  from .serialisation import tree_deserialise_leaves, tree_serialise_leaves
  from .tree import tree_at, tree_equal, tree_inference
  from .update import apply_updates
  from .vmap_pmap import filter_pmap, filter_vmap
  
  
  __version__ = "0.5.5"
  
new_contents: from . import experimental, nn
  from .eval_shape import filter_eval_shape
  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 .pretty_print import tree_pformat
  from .serialisation import tree_deserialise_leaves, tree_serialise_leaves
  from .tree import tree_at, tree_equal, tree_inference
  from .update import apply_updates
  from .vmap_pmap import filter_pmap, filter_vmap
  
  
  __version__ = "0.5.6"
  
subject: Bump version