ml-stable-diffusion/mlx/docs/src/python/nn/module.rst

Module
======

.. currentmodule:: mlx.nn

.. autoclass:: Module

   .. rubric:: Attributes

   .. autosummary::
      :toctree: _autosummary
   
      Module.training
      Module.state
   
   .. rubric:: Methods

   .. autosummary::
      :toctree: _autosummary
   
      Module.apply
      Module.apply_to_modules
      Module.children
      Module.eval
      Module.filter_and_map
      Module.freeze
      Module.leaf_modules
      Module.load_weights
      Module.modules
      Module.named_modules
      Module.parameters
      Module.save_weights
      Module.set_dtype
      Module.train
      Module.trainable_parameters
      Module.unfreeze
      Module.update
      Module.update_modules