mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 18:11:15 +08:00
39 lines
700 B
ReStructuredText
39 lines
700 B
ReStructuredText
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
|