namedtuple_utils

jaxns.internals.namedtuple_utils

Module Contents

isinstance_namedtuple(obj)[source]

Check if object is a namedtuple.

Parameters:

obj – object

Returns:

bool

Return type:

bool

issubclass_namedtuple(cls)[source]

Check if the type object is a subclass of a namedtuple.

serialise_namedtuple(obj)[source]
deserialise_namedtuple(obj)[source]
serialise_ndarray(obj)[source]
deserialise_ndarray(obj)[source]
serialise_jax_ndarray(obj)[source]
deserialise_jax_ndarray(obj)[source]