| .. |
|
3rdparty
|
jagrit's commit files
|
2023-11-29 10:52:08 -08:00 |
|
backend
|
CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307)
|
2024-08-08 15:18:02 -07:00 |
|
distributed
|
Add docs for the distributed namespace (#1184)
|
2024-06-06 11:37:00 -07:00 |
|
io
|
Add gemv masked to JIT plus some fixes (#1310)
|
2024-08-07 13:38:07 -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 with multi-output primitives (#1274)
|
2024-07-23 06:34:18 -07:00 |
|
array.h
|
minor fixes (#1194)
|
2024-06-12 22:06:49 -07:00 |
|
CMakeLists.txt
|
Einsum (#1269)
|
2024-07-25 09:36:44 -07:00 |
|
compile_impl.h
|
Ios compile (#784)
|
2024-03-04 20:02:26 -08:00 |
|
compile.cpp
|
Fixes segfault when compiling checkpointed functions (#1235)
|
2024-06-26 16:14:45 -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
|
minor fixes (#1194)
|
2024-06-12 22:06:49 -07:00 |
|
dtype.cpp
|
Add gemv masked to JIT plus some fixes (#1310)
|
2024-08-07 13:38:07 -07:00 |
|
dtype.h
|
Add gemv masked to JIT plus some fixes (#1310)
|
2024-08-07 13:38:07 -07:00 |
|
einsum.cpp
|
Einsum (#1269)
|
2024-07-25 09:36:44 -07:00 |
|
einsum.h
|
Einsum (#1269)
|
2024-07-25 09:36:44 -07:00 |
|
event.h
|
minor fixes (#1194)
|
2024-06-12 22:06:49 -07:00 |
|
fast_primitives.h
|
Fused Affine Quantize/Dequantize ops (#1282)
|
2024-07-29 15:11:38 -07:00 |
|
fast.cpp
|
Fused Affine Quantize/Dequantize ops (#1282)
|
2024-07-29 15:11:38 -07:00 |
|
fast.h
|
Fused Affine Quantize/Dequantize ops (#1282)
|
2024-07-29 15:11:38 -07:00 |
|
fft.cpp
|
Feature complete Metal FFT (#1102)
|
2024-06-06 12:57:25 -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
|
CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307)
|
2024-08-08 15:18:02 -07:00 |
|
linalg.h
|
CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307)
|
2024-08-08 15:18:02 -07:00 |
|
mlx.h
|
Einsum (#1269)
|
2024-07-25 09:36:44 -07:00 |
|
ops.cpp
|
Revert variance to be numerically stable (#1314)
|
2024-08-08 13:35:02 -07:00 |
|
ops.h
|
Add "edge" mode to mx.pad (#1309)
|
2024-08-06 11:23:10 -07:00 |
|
primitives.cpp
|
Add "edge" mode to mx.pad (#1309)
|
2024-08-06 11:23:10 -07:00 |
|
primitives.h
|
CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307)
|
2024-08-08 15:18:02 -07:00 |
|
random.cpp
|
Einsum (#1269)
|
2024-07-25 09:36:44 -07:00 |
|
random.h
|
Implement sampling from laplace distribution. (#1279)
|
2024-07-24 15:15:37 +02:00 |
|
scheduler.cpp
|
Add synchronize function (#1006)
|
2024-04-22 08:25:46 -07:00 |
|
scheduler.h
|
minor fixes (#1194)
|
2024-06-12 22:06:49 -07:00 |
|
stream.h
|
Add synchronize function (#1006)
|
2024-04-22 08:25:46 -07:00 |
|
transforms_impl.h
|
Custom transforms (#1246)
|
2024-07-10 18:00:01 -07:00 |
|
transforms.cpp
|
Add vmap to scatter (#1200)
|
2024-08-05 20:12:27 -07:00 |
|
transforms.h
|
Custom transforms (#1246)
|
2024-07-10 18:00:01 -07:00 |
|
utils.cpp
|
Shared events for synchronization + async eval (#998)
|
2024-04-17 06:16:02 -07:00 |
|
utils.h
|
Add gemv masked to JIT plus some fixes (#1310)
|
2024-08-07 13:38:07 -07:00 |
