6 Commits

Author	SHA1	Message	Date
Awni Hannun	4f46e9c997	More fixes for arrays with large sizes (#1405) ... * compile works for big arrays when contiguous * style * nits in docs * a bunch more stuff * update jit * update jit * use constant for shapes and strides and remove elem_to_loc overload * use kernel instantiation * docs nits * update binary and ternary * comments	2024-09-17 12:46:31 -07:00
Nikhil Mehta	0b7d71fd2f	Add softmin, hardshrink, hardtanh (#1180) ... --------- Co-authored-by: Nikhil Mehta <nikmehta@tesla.com>	2024-06-04 15:48:18 -07:00
Awni Hannun	fe1dabf272	Fix compile with non standard types (#745) ... * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-02-26 19:28:53 -08:00
Andre Slavescu	d3a9005454	Softshrink mapping + op (#552) ... * Added Softshrink mapping + op * formatting * docs + nits in docstring --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-01-30 12:56:28 -08:00
AtomicVar	4a9012cba0	Sort some APIs docs by names (a-z) (#472)	2024-01-16 19:37:50 -08:00
Awni Hannun	ee0c2835c5	Docs updates (#198) ... Reorganize NN docs + a few other tidbits.	2023-12-17 13:20:55 -08:00