mlx/mlx
Angelos Katharopoulos d8fabaa12b
Split multi output (#461)
* Multi-output split primitive
* Add the multi-output split to the ArrayIterator
* Add some grad tests for split
2024-01-16 13:33:55 -08:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Split multi output (#461) 2024-01-16 13:33:55 -08:00
io Fix packaging of gguflib (#435) 2024-01-11 13:56:03 -08:00
types Spelling (#342) 2024-01-01 21:08:17 -08:00
allocator.cpp Memory allocation (#292) 2024-01-02 11:59:19 -08:00
allocator.h Memory allocation (#292) 2024-01-02 11:59:19 -08:00
array.cpp Split multi output (#461) 2024-01-16 13:33:55 -08:00
array.h Split multi output (#461) 2024-01-16 13:33:55 -08:00
CMakeLists.txt Safetensor support (#215) 2023-12-27 02:06:55 -08:00
device.cpp copyright + ack 2023-11-30 11:12:53 -08:00
device.h copyright + ack 2023-11-30 11:12:53 -08:00
dtype.cpp copyright + ack 2023-11-30 11:12:53 -08:00
dtype.h random generation fix (#80) 2023-12-08 10:40:57 -08:00
fft.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
fft.h copyright + ack 2023-11-30 11:12:53 -08:00
graph_utils.cpp Fix stub generation, change graph exporting for arrows to go to outputs (#455) 2024-01-14 14:06:16 -08:00
graph_utils.h copyright + ack 2023-11-30 11:12:53 -08:00
linalg.cpp linalg.norm (#187) 2023-12-26 19:42:04 -08:00
linalg.h linalg.norm (#187) 2023-12-26 19:42:04 -08:00
mlx.h linalg.norm (#187) 2023-12-26 19:42:04 -08:00
ops.cpp Split multi output (#461) 2024-01-16 13:33:55 -08:00
ops.h implemented isposinf and isneginf in one PR (#470) 2024-01-16 06:48:07 -08:00
primitives.cpp Split multi output (#461) 2024-01-16 13:33:55 -08:00
primitives.h Split multi output (#461) 2024-01-16 13:33:55 -08:00
random.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
random.h feat: add time based seed to random.h (#457) 2024-01-16 07:32:28 -08:00
scheduler.cpp copyright + ack 2023-11-30 11:12:53 -08:00
scheduler.h move all ObjC (via metal-cpp) interaction until post static initializers (#370) 2024-01-04 16:12:00 -08:00
stream.h copyright + ack 2023-11-30 11:12:53 -08:00
transforms_impl.h Removes the retain_graph flag (#385) 2024-01-07 15:16:51 -08:00
transforms.cpp Use a dummy primitive to only sync with one output (#453) 2024-01-14 14:09:40 -08:00
transforms.h Removes the retain_graph flag (#385) 2024-01-07 15:16:51 -08:00
utils.cpp implemented Flatten Module (#149) 2023-12-16 21:54:37 -08:00
utils.h Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00