14 Commits

Author	SHA1	Message	Date
kaixih	0d387e0839	Update jax doc sdpa	2024-07-15 17:30:54 +00:00
rajasekharporeddy	0a6827f4e3	Add sparse_sigmoid to jax.nn docs	2024-05-08 12:23:01 +05:30
carlosgmartin	f0314c70e8	Add jax.nn.mish.	2024-04-03 16:37:07 -04:00
Matteo Hessel	c94ea147f2	Add sparseplus activation to jax.nn. ... PiperOrigin-RevId: 616087452	2024-03-15 04:40:38 -07:00
carlosgmartin	9f8e1bc34a	Add nn.squareplus.	2023-11-14 23:52:41 -05:00
Jake VanderPlas	81e627d5bd	DOC: make API doc titles more uniform	2023-01-18 10:59:42 -08:00
Matthew Johnson	6a3d2a0dde	update docs to point to jax.nn.standardize ... Fixes #12909	2022-10-20 21:25:37 -07:00
Alex Alemi	44899c907f	Alphabetize docs.	2020-10-15 11:10:56 -04:00
Alex Alemi	f7d4063e55	Remove expit, add logsumexp to docs	2020-10-15 11:06:18 -04:00
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