Awni Hannun e142aaf8a1 Option for precise softmax (#953) ... * precise softmax * Add an equivalency check * Make the threadgroup memory definition fixed * precise cpu softmax * precise option on cpu * remove print --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>		2024-04-04 08:32:35 -07:00
..
mlx	fix a couple bugs (#952)	2024-04-02 12:07:41 -07:00
src	Option for precise softmax (#953)	2024-04-04 08:32:35 -07:00
tests	Option for precise softmax (#953)	2024-04-04 08:32:35 -07:00