Awni Hannun 4758c8baa1 Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777) ...

* start to cleanup/unify accelerate and common back-ends

* more progress

* simplify

* add half type and allow infs in simd exp

* unify softmax + quantized, more dispatches to simd quantized mm

* add sin/cos, use simd in vector-scalar ops

* faster CPU vectorize quant

* faster erf/erfinv

2025-01-29 14:34:49 -08:00

..

array.cpp

Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777)

2025-01-29 14:34:49 -08:00

buffer.h

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

CMakeLists.txt

allow build python lib without specifying path (#1799)

2025-01-27 11:22:35 -08:00

constants.cpp

Array api (#1289)

2024-07-26 10:40:49 -07:00

convert.cpp

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

convert.h

Bump nanobind to 2.4 + fix (#1710)

2024-12-17 10:57:54 -08:00

device.cpp

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

distributed.cpp

Ring distributed backend (#1784)

2025-01-27 22:15:01 -08:00

export.cpp

Optionally specify names for arrays when exporting (#1749)

2025-01-06 13:07:46 -08:00

fast.cpp

Allow boolean mask in sdpa (#1753)

2025-01-06 16:57:07 -08:00

fft.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

indexing.cpp

Dynamic slicing (#1741)

2025-01-07 14:02:16 -08:00

indexing.h

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

linalg.cpp

Use @ matrix multiplication syntax to document matrix-matrix multiplication (#1793)

2025-01-25 16:02:36 -08:00

load.cpp

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

load.h

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

metal.cpp

track resource limit and throw if exceeded (#1718)

2024-12-18 18:45:58 -08:00

mlx.cpp

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

ops.cpp

Fix broadcast_arrays python sig (#1763)

2025-01-10 12:33:26 -08:00

random.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

stream.cpp

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

transforms.cpp

Limit grad recursion depth by not recursing through non-grad inputs (#1764)

2025-01-14 14:33:18 -08:00

trees.cpp

Limit grad recursion depth by not recursing through non-grad inputs (#1764)

2025-01-14 14:33:18 -08:00

trees.h

Limit grad recursion depth by not recursing through non-grad inputs (#1764)

2025-01-14 14:33:18 -08:00

utils.cpp

Bump nanobind to 2.4 + fix (#1710)

2024-12-17 10:57:54 -08:00

utils.h

Bump nanobind to 2.4 + fix (#1710)

2024-12-17 10:57:54 -08:00