Back to Mlx

Module

docs/src/python/nn/module.rst

0.31.2699 B
Original Source

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