Alex Barron 95d11bda06 Fix NumPy 2.0 pickle test (#1221) ...

* fix numpy version <2 temporarily

* typo

* better fix

* Fix just for bfloat16

---------

Co-authored-by: Alex Barron <abarron22@apple.com>

2024-06-23 05:47:22 -07:00

..

array.cpp

Add view op (#1179)

2024-06-04 08:05:27 -07:00

buffer.h

Fix buffer protocol buffer size designation (#1010)

2024-04-19 06:06:13 -07:00

CMakeLists.txt

Comms (#1097)

2024-05-23 17:04:02 -07:00

constants.cpp

Switch to nanobind (#839)

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

convert.cpp

Fix NumPy 2.0 pickle test (#1221)

2024-06-23 05:47:22 -07:00

convert.h

list based indexing (#1150)

2024-05-22 15:52:05 -07:00

device.cpp

Switch to nanobind (#839)

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

distributed.cpp

Add docs for the distributed namespace (#1184)

2024-06-06 11:37:00 -07:00

fast.cpp

Metal shaders for memory efficient self attention on large sequences (#964)

2024-06-03 09:16:19 -07:00

fft.cpp

Switch to nanobind (#839)

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

indexing.cpp

list based indexing (#1150)

2024-05-22 15:52:05 -07:00

indexing.h

Switch to nanobind (#839)

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

linalg.cpp

Some fixes in docs (#1141)

2024-05-20 11:51:47 -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

Reset peak memory (#1074)

2024-05-03 17:12:51 -07:00

mlx.cpp

Comms (#1097)

2024-05-23 17:04:02 -07:00

ops.cpp

Add view op (#1179)

2024-06-04 08:05:27 -07:00

random.cpp

Implementation of mlx.random.multivariate_normal (#502) (#877)

2024-04-09 13:50:12 -07:00

stream.cpp

Cpp docs (#1036)

2024-04-26 12:56:05 -07:00

transforms.cpp

add keyword positonal (#1081)

2024-05-06 07:18:49 -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.cpp

fix conversion to array (#1070)

2024-05-06 16:02:49 -07:00

utils.h

fix conversion to array (#1070)

2024-05-06 16:02:49 -07:00