jax_dna.input.tree

Functions for saving and loading pytrees.

Functions

`save_pytree`(→ None)	Save a pytree to a file.
`load_pytree`(→ jax_dna.utils.types.PyTree)	Load a pytree to a file.

jax_dna.input.tree.save_pytree(data: jax_dna.utils.types.PyTree, filename: jax_dna.utils.types.PathOrStr) → None[source]: Save a pytree to a file.

jax_dna.input.tree.load_pytree(filename: jax_dna.utils.types.PathOrStr) → jax_dna.utils.types.PyTree[source]: Load a pytree to a file.