mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 19:36:04 +00:00
Code Issues Packages Projects Releases Wiki Activity
4,979 Commits 508 Branches 263 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
Peter Hawkins
a2fb798dc3
Rename swish to silu, add swish as an alias to silu. (#3673) 2020-07-06 18:08:16 -04:00
Peter Hawkins
3e87e8f960
Add relu6, hard_swish, and hard_sigmoid to docs. (#2886) 2020-04-29 14:59:23 -04:00
Peter Hawkins
7629c5aab4
Add some missing functions to documents. (#2615) 2020-04-06 12:39:28 -04:00
Peter Hawkins
f0d9333379
Document functions in jax.nn. (#1795) 2019-12-02 14:21:10 -05:00
James Bradbury
cc49d8b325 add docs for jax.nn 2019-08-29 18:15:36 -07:00