mlx/mlx
2024-05-17 12:31:59 -07:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Implemented Cholesky on CPU (#1119) 2024-05-17 12:31:59 -07:00
io Add compiler flags to disable safetensors and gguf (#1098) 2024-05-09 17:39:44 -07:00
types feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
allocator.cpp Memory allocation (#292) 2024-01-02 11:59:19 -08:00
allocator.h feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
array.cpp Fix leak for multi-output primitives which are never detached (#1059) 2024-05-01 07:31:45 -07:00
array.h Fix leak for multi-output primitives which are never detached (#1059) 2024-05-01 07:31:45 -07:00
CMakeLists.txt No CPU option for binary minimization (#1105) 2024-05-13 16:08:11 -07:00
compile_impl.h Ios compile (#784) 2024-03-04 20:02:26 -08:00
compile.cpp Add conjugate operator (#1100) 2024-05-10 07:22:20 -07:00
compile.h Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00
device.cpp copyright + ack 2023-11-30 11:12:53 -08:00
device.h feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
dtype.cpp Remove unnecessary string copies (#891) 2024-03-28 13:14:59 -07:00
dtype.h feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
event.h feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
fast_primitives.h feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
fast.cpp Option for precise softmax (#953) 2024-04-04 08:32:35 -07:00
fast.h Adds mx.fast.layer_norm (#870) 2024-03-21 13:55:51 -07:00
fft.cpp Don't use make_unique to create shared_ptr (#902) 2024-03-27 06:13:29 -07:00
fft.h Remove duplicate defines of StreamOrDevice and is_big_endian (#892) 2024-03-26 15:15:11 -07:00
graph_utils.cpp Remove unnecessary string copies (#891) 2024-03-28 13:14:59 -07:00
graph_utils.h Add missing && when forwarding args (#894) 2024-03-25 14:55:54 -07:00
io.h Add compiler flags to disable safetensors and gguf (#1098) 2024-05-09 17:39:44 -07:00
linalg.cpp Implemented Cholesky on CPU (#1119) 2024-05-17 12:31:59 -07:00
linalg.h Implemented Cholesky on CPU (#1119) 2024-05-17 12:31:59 -07:00
mlx.h Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
ops.cpp Block sparse qmm (#1124) 2024-05-16 15:24:14 -07:00
ops.h Block sparse qmm (#1124) 2024-05-16 15:24:14 -07:00
primitives.cpp Block sparse qmm (#1124) 2024-05-16 15:24:14 -07:00
primitives.h Implemented Cholesky on CPU (#1119) 2024-05-17 12:31:59 -07:00
random.cpp Implementation of mlx.random.multivariate_normal (#502) (#877) 2024-04-09 13:50:12 -07:00
random.h Implementation of mlx.random.multivariate_normal (#502) (#877) 2024-04-09 13:50:12 -07:00
scheduler.cpp Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00
scheduler.h fix order device -> scheduler (#1039) 2024-04-26 13:46:41 -07:00
stream.h Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00
transforms_impl.h Use uintptr_t instead of size_t to store funtion id (#916) 2024-03-28 06:37:59 -07:00
transforms.cpp improve error messaging in eval (#1101) 2024-05-10 10:04:07 -07:00
transforms.h Shared events for synchronization + async eval (#998) 2024-04-17 06:16:02 -07:00
utils.cpp Shared events for synchronization + async eval (#998) 2024-04-17 06:16:02 -07:00
utils.h Remove duplicate defines of StreamOrDevice and is_big_endian (#892) 2024-03-26 15:15:11 -07:00