Angelos Katharopoulos e7f5059fe4 Support for quantized matmul with w and w^T (#349) ...

* Add the metal qvm implementation
* Add qmm_n
* Add gradient wrt to input for quantized_matmul

2024-01-03 14:22:36 -08:00

..

array.cpp

Spelling (#342)

2024-01-01 21:08:17 -08:00

CMakeLists.txt

linalg.norm (#187)

2023-12-26 19:42:04 -08:00

device.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

fft.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

indexing.cpp

Indexing bug (#233)

2023-12-20 10:44:01 -08:00

indexing.h

copyright + ack

2023-11-30 11:12:53 -08:00

linalg.cpp

linalg.norm (#187)

2023-12-26 19:42:04 -08:00

load.cpp

Fix load compilation (#298)

2023-12-27 06:20:45 -08:00

load.h

Safetensor support (#215)

2023-12-27 02:06:55 -08:00

metal.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

mlx.cpp

linalg.norm (#187)

2023-12-26 19:42:04 -08:00

ops.cpp

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

stream.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

transforms.cpp

Corrected the example of value_and_grad (#274)

2023-12-23 11:06:38 -08:00

utils.h

Mlx array accessor (#128)

2023-12-11 13:42:55 -08:00