mlx.nn.Module.apply_to_modules

Module.apply_to_modules(apply_fn: Callable[[str, Module], Any])

Apply a function to all the modules in this instance (including this instance).

Parameters:

apply_fn (Callable) – The function to apply to the modules.