mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
479051ce1c
* add numeric type hierarchy and issubdtype as well as a set_dtype method to nn.Module with predicate
numeric type hierarchy and issubtype is compatible to the [numpy hierarchy](220f0ab2c5/numpy/_core/numerictypes.py (L42)).
Closes #285.
* nits in docs
* unify type category checking
* nits in docs
* nits in docs
* more docs nits
* fix callable type
---------
Co-authored-by: Awni Hannun <awni@apple.com>
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
|