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