mlx/mlx
mutexuan d8f41a5c0f
support python mlx.array creation from list of mlx.array's (#325)
* support python mlx.array creation from list of mlx.array's

* include bfloat16 in UT

* refactor so that sub array made of all python primitive types gets initialized by fill_vector

* address PR comment: arr.shape().size() -> arr.ndim()

* address PR comment: get back Dtype constness and let stack to handle type promotions automatically
2024-01-04 18:53:33 -08:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend bump pre commit and fix format (#373) 2024-01-04 16:28:52 -08:00
io Spelling (#342) 2024-01-01 21:08:17 -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 Spelling (#342) 2024-01-01 21:08:17 -08:00
array.h add base kwarg to rope (#186) 2023-12-15 16:47:59 -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 copyright + ack 2023-11-30 11:12:53 -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 support python mlx.array creation from list of mlx.array's (#325) 2024-01-04 18:53:33 -08:00
ops.h Support for quantized matmul with w and w^T (#349) 2024-01-03 14:22:36 -08:00
primitives.cpp Support for quantized matmul with w and w^T (#349) 2024-01-03 14:22:36 -08:00
primitives.h Support for quantized matmul with w and w^T (#349) 2024-01-03 14:22:36 -08:00
random.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
random.h Spelling (#342) 2024-01-01 21:08:17 -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 format 2023-11-30 11:50:50 -08:00
transforms.cpp copyright + ack 2023-11-30 11:12:53 -08:00
transforms.h Spelling (#342) 2024-01-01 21:08:17 -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