nicolov 8c9f0278b9 Add vmap to scatter (#1200) ...

* Add vmap to scatter

* updates

* vmap updates + a few more tests

* bug fix

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-08-05 20:12:27 -07:00

..

3rdparty

jagrit's commit files

2023-11-29 10:52:08 -08:00

backend

add bfloat conv for windograd (#1306)

2024-08-05 15:51:13 -07:00

distributed

Add docs for the distributed namespace (#1184)

2024-06-06 11:37:00 -07:00

io

fix gguf (#1273)

2024-07-18 07:35:35 -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

Remove unnecessary string copies (#891)

2024-03-28 13:14:59 -07:00

dtype.h

minor fixes (#1194)

2024-06-12 22:06:49 -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

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

Einsum (#1269)

2024-07-25 09:36:44 -07:00

ops.cpp

Add vmap to scatter (#1200)

2024-08-05 20:12:27 -07:00

ops.h

[Issue #1187] Add nan_to_num function initial attempt (#1247)

2024-07-25 09:57:37 -07:00

primitives.cpp

Add vmap to scatter (#1200)

2024-08-05 20:12:27 -07:00

primitives.h

Add vmap to scatter (#1200)

2024-08-05 20:12:27 -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

Fast Hadamard Transform (#1249)

2024-07-09 20:39:01 -07:00