| .. |
|
reduction
|
Check edge case handling in row reduce med kernel (#858)
|
2024-03-20 11:37:58 -07:00 |
|
steel
|
No copy gems (#801)
|
2024-03-12 13:13:41 -07:00 |
|
arange.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
arg_reduce.metal
|
Fix threadgroup memory in arg reduce (#723)
|
2024-02-21 19:42:16 -08:00 |
|
atomic.h
|
Fix eye for larger matrices (#463)
|
2024-01-16 00:51:24 -08:00 |
|
bf16_math.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
bf16.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
binary_two.metal
|
Remainder negative numerator bug fixed (#641)
|
2024-02-09 16:49:14 -08:00 |
|
binary.h
|
Remainder negative numerator bug fixed (#641)
|
2024-02-09 16:49:14 -08:00 |
|
binary.metal
|
Fix logsumexp edge case (#740)
|
2024-02-25 08:39:55 -08:00 |
|
CMakeLists.txt
|
Adds mx.fast.layer_norm (#870)
|
2024-03-21 13:55:51 -07:00 |
|
compiled_preamble.h
|
Implement the 'where' primitive for conditional selection (#664)
|
2024-02-22 15:10:48 -08:00 |
|
complex.h
|
Remainder negative numerator bug fixed (#641)
|
2024-02-09 16:49:14 -08:00 |
|
conv.metal
|
Convolution update (#651)
|
2024-02-28 20:11:16 -08:00 |
|
copy.metal
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -07:00 |
|
defines.h
|
Fast RMS Norm (#862)
|
2024-03-21 07:20:54 -07:00 |
|
erf.h
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
gather.metal
|
Update gather and scatter to not use Argument Encoder (#683)
|
2024-02-14 13:42:13 -08:00 |
|
gemv.metal
|
No copy gems (#801)
|
2024-03-12 13:13:41 -07:00 |
|
indexing.h
|
Update gather and scatter to not use Argument Encoder (#683)
|
2024-02-14 13:42:13 -08:00 |
|
layer_norm.metal
|
Implement vjps for some primitives in the fast namespace (#883)
|
2024-03-26 16:35:34 -07:00 |
|
quantized.metal
|
Fix OOB read in qmv when non-divisible by blocksize (#917)
|
2024-03-27 22:18:35 -07:00 |
|
random.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
rms_norm.metal
|
Implement vjps for some primitives in the fast namespace (#883)
|
2024-03-26 16:35:34 -07:00 |
|
rope.metal
|
Implement vjps for some primitives in the fast namespace (#883)
|
2024-03-26 16:35:34 -07:00 |
|
scaled_dot_product_attention_params.h
|
Fast Inference SDPA op (#735)
|
2024-03-04 21:06:11 -08:00 |
|
scaled_dot_product_attention.metal
|
Compile stride bug (#812)
|
2024-03-11 06:31:31 -07:00 |
|
scan.metal
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
scatter.metal
|
fix scatter (#821)
|
2024-03-12 11:42:07 -07:00 |
|
softmax.metal
|
Fast RMS Norm (#862)
|
2024-03-21 07:20:54 -07:00 |
|
sort.metal
|
Spelling (#342)
|
2024-01-01 21:08:17 -08:00 |
|
ternary.h
|
Implement the 'where' primitive for conditional selection (#664)
|
2024-02-22 15:10:48 -08:00 |
|
ternary.metal
|
Fix logsumexp edge case (#740)
|
2024-02-25 08:39:55 -08:00 |
|
unary.h
|
Fix logsumexp edge case (#740)
|
2024-02-25 08:39:55 -08:00 |
|
unary.metal
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
utils.h
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -07:00 |
