jaxdem.utils.serialization

Functions

decode_callable(path)	Import a callable from a dotted path string.
encode_callable(fn)	Return a dotted path like 'jax._src.nn.functions.gelu'.

jaxdem.utils.serialization.encode_callable(fn: Callable) → str

Return a dotted path like ‘jax._src.nn.functions.gelu’.

jaxdem.utils.serialization.decode_callable(path: str) → Callable

Import a callable from a dotted path string.