penzai.toolshed
===============

.. automodule:: penzai.toolshed

   
   
   

   
   
   

   
   
   

   
   
   



.. rubric:: Submodules

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

   penzai.toolshed.auto_nmap
   penzai.toolshed.basic_training
   penzai.toolshed.gradient_checkpointing
   penzai.toolshed.isolate_submodel
   penzai.toolshed.jit_wrapper
   penzai.toolshed.lora
   penzai.toolshed.model_rewiring
   penzai.toolshed.patch_ipdb
   penzai.toolshed.save_intermediates
   penzai.toolshed.sharding_util
   penzai.toolshed.token_visualization
   penzai.toolshed.unflaxify