L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

implement-batch-norm-layer (#217)

Browse Source 
- Add batch normalization layer

---------

Co-authored-by: Robert McCraith <[email protected]>
Co-authored-by: Awni Hannun <[email protected]>
This commit is contained in:
__mo_san__
2023-12-25 07:32:53 -08:00
committed by GitHub
co-authored by Robert McCraith Awni Hannun
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
docs/src/python/nn/layers.rst
+1
View File
@@ -20,6 +20,7 @@ Layers
Linear
Conv1d
Conv2d
BatchNorm
LayerNorm
RMSNorm
GroupNorm