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 (
    default_deserialise_filter_spec,
    default_serialise_filter_spec,
    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.7.0"

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 (
    default_deserialise_filter_spec,
    default_serialise_filter_spec,
    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.7.1"

bump version