Awni Hannun e6a7ab9675 non square qr (#1783)

2025-01-21 14:07:47 -08:00

..

arange.h

copyright + ack

2023-11-30 11:12:53 -08:00

arg_reduce.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

binary_two.h

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

binary.cpp

smaller CPU binary (#1203)

2024-06-14 09:46:55 -07:00

binary.h

Explicit parentheses around some logical operators (#1732)

2024-12-24 07:02:20 -08:00

cholesky.cpp

Eigenvalues and eigenvectors (#1334)

2024-10-22 12:18:48 -07:00

CMakeLists.txt

Make mx.compile work on Windows (#1697)

2024-12-24 07:02:33 -08:00

common.cpp

Dynamic broadcasting for shapeless compile/export (#1722)

2025-01-09 11:04:24 -08:00

compiled_cpu.cpp

Include command stdout in error message (#1756)

2025-01-08 07:17:03 -08:00

compiled_nocpu.cpp

Ios compile (#784)

2024-03-04 20:02:26 -08:00

compiled_preamble.h

Refactor CPU compile preamble (#708)

2024-02-19 06:12:53 -08:00

compiled.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

compiled.h

Simplify removes no-ops from the tape (#1759)

2025-01-09 11:23:19 -08:00

conv.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

copy.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

copy.h

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

default_primitives.cpp

Dynamic broadcasting for shapeless compile/export (#1722)

2025-01-09 11:04:24 -08:00

eigh.cpp

Eigenvalues and eigenvectors (#1334)

2024-10-22 12:18:48 -07:00

erf.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

fft.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

hadamard.cpp

Fast Hadamard Transform (#1249)

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

hadamard.h

Fast Hadamard Transform (#1249)

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

indexing.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

inverse.cpp

Eigenvalues and eigenvectors (#1334)

2024-10-22 12:18:48 -07:00

jit_compiler.cpp

Include command stdout in error message (#1756)

2025-01-08 07:17:03 -08:00

jit_compiler.h

Include command stdout in error message (#1756)

2025-01-08 07:17:03 -08:00

lapack.h

Use correct complex type for MSVC (#1660)

2024-12-07 18:25:22 -08:00

load.cpp

Async load (#1372)

2024-08-28 14:21:55 -07:00

load.h

Async load (#1372)

2024-08-28 14:21:55 -07:00

make_compiled_preamble.ps1

Make mx.compile work on Windows (#1697)

2024-12-24 07:02:33 -08:00

make_compiled_preamble.sh

Fix x86 tests (#1691)

2024-12-11 07:47:18 -08:00

masked_mm.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

ops.h

Do not pass integers to isnan (#1664)

2024-12-07 18:26:23 -08:00

primitives.cpp

Dynamic slicing (#1741)

2025-01-07 14:02:16 -08:00

qrf.cpp

non square qr (#1783)

2025-01-21 14:07:47 -08:00

quantized.cpp

3 and 6 bit quantization (#1613)

2024-11-22 10:22:13 -08:00

reduce_utils.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

reduce.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

reduce.h

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

scan.cpp

Fix scan (#1188)

2024-06-05 14:21:58 -07:00

select.cpp

Implement the 'where' primitive for conditional selection (#664)

2024-02-22 15:10:48 -08:00

slicing.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

slicing.h

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

softmax.cpp

Option for precise softmax (#953)

2024-04-04 08:32:35 -07:00

sort.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

svd.cpp

Eigenvalues and eigenvectors (#1334)

2024-10-22 12:18:48 -07:00

ternary.h

Allow boolean mask in sdpa (#1753)

2025-01-06 16:57:07 -08:00

threefry.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

threefry.h

copyright + ack

2023-11-30 11:12:53 -08:00

unary.h

Real and Imag (#1490)

2024-10-15 16:23:15 -07:00

utils.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

utils.h

More shape type (#1705)

2024-12-19 08:08:20 -08:00