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

Fix compile with non standard types (#745)

Browse Source 
* refactor tree utils

* fix compile + tree code refactor

* Add an extra test

* add a few missing activations to docs

* hash structure

* Encode the full argument structure

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
This commit is contained in:
Awni Hannun
2024-02-26 19:28:53 -08:00
committed by GitHub
parent 08226ab491
commit fe1dabf272
8 changed files with 438 additions and 282 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
docs/src/python/nn/functions.rst
View File

@@ -12,13 +12,24 @@ simple functions.
:toctree: _autosummary_functions
:template: nn-module-template.rst
elu
gelu
gelu_approx
gelu_fast_approx
glu
hardswish
leaky_relu
log_sigmoid
log_softmax
mish
prelu
relu
relu6
selu
softshrink
sigmoid
silu
softmax
softplus
softshrink
step
tanh