mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-18 23:21:16 +08:00
6b4b30e3fc
* initial commit: constant, normal, uniform * identity, glorot and he initializers * docstrings * rm file * nits * nits * nits * testing suite * docs * nits in docs * more docs * remove unused template * rename packakge to nn.innit * docs, receptive field * more docs --------- Co-authored-by: Awni Hannun <awni@apple.com>
6 lines
132 B
Python
6 lines
132 B
Python
# Copyright © 2023 Apple Inc.
|
|
|
|
from mlx.nn import init, losses
|
|
from mlx.nn.layers import *
|
|
from mlx.nn.utils import value_and_grad
|