6 Commits

Author	SHA1	Message	Date
Cheng	79071bfba4	Fix out-of-bounds default value in logsumexp/softmax (#2213)	2025-05-21 07:25:16 -07:00
Awni Hannun	c41f7565ed	fix softmax / logsumexp (#2042)	2025-04-03 08:32:59 -07:00
Awni Hannun	de5f38fd48	Custom logsumexp (#2028) ... * initial custom logsumexp * more tests * comments + fix	2025-03-31 07:36:55 -07:00
Awni Hannun	1fa0d20a30	consistently handle all -inf in softmax (#1470)	2024-10-08 09:54:02 -07:00
Awni Hannun	40b6d67333	Fixes for large arrays with a few ops (#1299) ... * fixes for large arrays with a few ops * fix bug * fix all of copy	2024-07-30 17:18:39 -07:00
Awni Hannun	0189ab6ab6	More jitting (#1132) ... * docs + circle min size build * jit scan, arange, softmax * add sort * jit reductions * remove print * fix deps * clean includes / nits	2024-05-23 16:23:44 -07:00