Matthew Johnson 2678a4647a

omnistaging on by default (#4038)

2020-09-15 08:06:46 -07:00

..

__init__.py

Rename swish to silu, add swish as an alias to silu. (#3673)

2020-07-06 18:08:16 -04:00

functions.py

omnistaging on by default (#4038)

2020-09-15 08:06:46 -07:00

initializers.py

Cleanup: avoid jnp.prod & np.prod on array shapes (#4086)

2020-08-18 10:17:38 -07:00