mlx.nn.Module.apply_to_modules# Module.apply_to_modules(apply_fn: Callable[[str, Module], Any]) → Module# Apply a function to all the modules in this instance (including this instance). Parameters: apply_fn (Callable) – The function to apply to the modules. Returns: The module instance after updating submodules.
