mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #21125 from rajasekharporeddy:test_branch1

Browse Source 
PiperOrigin-RevId: 640648819
This commit is contained in:
jax authors 2024-06-05 14:24:38 -07:00
commit 444a4c9110
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/jax.nn.rst
View File

@ -23,6 +23,7 @@ Activation functions
sigmoid
softplus
sparse_plus
sparse_sigmoid
soft_sign
silu
swish