3 Commits

Author	SHA1	Message	Date
Thomas Santerre	c5626b8271	Add support for "sign" on tensors (#2012) ... * add the sign unary operator * remove uneeded import * remove uneeded import * undo formatting * undo formatting * remove unnecessary redefintion * allow gradient to flow through for sign and round * fix cpu ops to ensure that negzero and positive zero are handled properly * clippy fixes * Properly avoid gradient tracking. * Use a branchless version. --------- Co-authored-by: laurent <laurent.mazare@gmail.com>	2024-04-04 22:32:47 +02:00
Thomas Santerre	5aebe53dd2	update dtypes checks for several metal operations (#2010)	2024-04-04 18:39:06 +02:00
Laurent Mazare	665da30487	Backend refactoring. (#1966) ... * Backend refactoring. * Metal tweaks. * Move the cudnn module.	2024-03-29 23:02:11 +01:00