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