zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-04 15:04:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

implement-batch-norm-layer (#217)

Browse Source 
- Add batch normalization layer

---------

Co-authored-by: Robert McCraith <mccraithrobert@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
__mo_san__
2023-12-25 16:32:53 +01:00
committed by GitHub
parent 9e6b8c9f48
commit a123c3c7d2
6 changed files with 267 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/src/python/nn/layers.rst
View File

@@ -20,6 +20,7 @@ Layers
Linear
Conv1d
Conv2d
BatchNorm
LayerNorm
RMSNorm
GroupNorm