mlx
mlx copied to clipboard
[Feature Request] Add the reset_parameters function of the builtin layers in nn.Module.
As a reset_parameters function of some layer usually is custom in PyTorch, while that is missing in MLX.