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"

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"
