diffusion\_models.models.nn
===========================

.. automodule:: diffusion_models.models.nn
  
   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree:                                          
   
      avg_pool_nd
      checkpoint
      conv_nd
      linear
      mean_flat
      normalization
      scale_module
      timestep_embedding
      update_ema
      zero_module
   
   

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree:                                          
      :template: custom-class-template.rst               
   
      CheckpointFunction
      GroupNorm32
      SiLU