4 Commits

Author	SHA1	Message	Date
chenyu	1692087db5	_one_hot_along_dim input needs to be int (#9179) ... * _one_hot_along_dim input needs to be int indexing and onehot compare with arange, and non-int dtype is likely a bug	2025-02-20 09:00:43 -05:00
SnakeOnex	0c49d38ba7	replace with tensor op (#3099)	2024-01-12 14:13:40 -05:00
chenyu	507e0afba0	fix onehot and jit in examples/transformer (#3073) ... trained to 0.999 in < 6 seconds on M1 Max consistently	2024-01-10 02:22:41 -05:00
George Hotz	0cbf6c1811	move things, clean up extra (#2292) ... * move things * idk why pylint needs that now * delete unused	2023-11-13 20:18:40 -08:00