penzai.experimental.v2.toolshed
===============================

.. automodule:: penzai.experimental.v2.toolshed

   
   
   

   
   
   

   
   
   

   
   
   



.. rubric:: Submodules

.. autosummary::
   :toctree:
   :template: pzmodule.rst
   :recursive:

   penzai.experimental.v2.toolshed.basic_training
   penzai.experimental.v2.toolshed.isolate_submodel
   penzai.experimental.v2.toolshed.jit_wrapper
   penzai.experimental.v2.toolshed.lora
   penzai.experimental.v2.toolshed.model_rewiring
   penzai.experimental.v2.toolshed.save_intermediates
   penzai.experimental.v2.toolshed.sharding_util
   penzai.experimental.v2.toolshed.unflaxify