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

Softshrink mapping + op (#552)

Browse Source 
* Added Softshrink mapping + op

* formatting

* docs + nits in docstring

---------

Co-authored-by: Awni Hannun <[email protected]>
This commit is contained in:
Andre Slavescu
2024-01-30 12:56:28 -08:00
committed by GitHub
co-authored by Awni Hannun
parent 3f7aba8498
commit d3a9005454
5 changed files with 70 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/src/python/nn/layers.rst
+1
View File
@@ -33,5 +33,6 @@ Layers
Sequential
SiLU
SinusoidalPositionalEncoding
Softshrink
Step
Transformer