mlx/python/src
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