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

..

array.cpp

Better exceptions in case of invalid operations on mlx.core.array (#910) (#926)

2024-04-02 21:11:24 -07:00

buffer.h

Remove unnecessary string copies (#891)

2024-03-28 13:14:59 -07:00

CMakeLists.txt

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

constants.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

convert.cpp

Ensure shape dimensions are within supported integer range (#566) (#704)

2024-03-25 13:29:45 -07:00

convert.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

device.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

fast.cpp

post nanobind docs fixes and some updates (#889)

2024-03-24 15:03:27 -07:00

fft.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

indexing.cpp

Indexing bug fix (#947)

2024-04-01 12:18:50 -07:00

indexing.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

linalg.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

load.cpp

Do not store iostream in shared_ptr (#872)

2024-03-22 06:54:45 -07:00

load.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

metal.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

mlx.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

ops.cpp

Option for precise softmax (#953)

2024-04-04 08:32:35 -07:00

random.cpp

post nanobind docs fixes and some updates (#889)

2024-03-24 15:03:27 -07:00

stream.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

transforms.cpp

Properly handle negative axes in python vmap (#944)

2024-04-02 18:07:23 -07:00

trees.cpp

Properly handle negative axes in python vmap (#944)

2024-04-02 18:07:23 -07:00

trees.h

Properly handle negative axes in python vmap (#944)

2024-04-02 18:07:23 -07:00

utils.h

Better exceptions in case of invalid operations on mlx.core.array (#910) (#926)

2024-04-02 21:11:24 -07:00