mbi.load

mbi.load(file: str | PathLike | IOBase) → Any[source]

Load a JAX pytree from .npz format.

Leaf values are returned as jax.Array so that JIT-compiled functions see consistent types.

Parameters:

file – Path string or readable binary file-like object.

Returns:

The reconstructed pytree.