jax_dna.input.tree
==================

.. py:module:: jax_dna.input.tree

.. autoapi-nested-parse::

   Functions for saving and loading pytrees.



Functions
---------

.. autoapisummary::

   jax_dna.input.tree.save_pytree
   jax_dna.input.tree.load_pytree


Module Contents
---------------

.. py:function:: save_pytree(data: jax_dna.utils.types.PyTree, filename: jax_dna.utils.types.PathOrStr) -> None

   Save a pytree to a file.


.. py:function:: load_pytree(filename: jax_dna.utils.types.PathOrStr) -> jax_dna.utils.types.PyTree

   Load a pytree to a file.