mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 09:21:16 +08:00
baf9fa5f42
* einsum initial * fix comma break * sum axis was wrong * small cleanups * python binding * changed bindings to resemble numpy * remove todo comment * comment changes * add count of operands/inputs * fail fast if operands list is empty * ignore comma if no output * einsum path matching numpy * getting somewhere with path * remove print * it passes the first test * moved einsum tests to seperate file * seperated einsum path * moved einsum naive * remove space from equation * fast fail if no operands passed * update tests and remove printf * small cleanup * some more cleanups * removed python helper file * ack * utilize std for finding min in vector * duplicate def * remove the tuple as it was unreadable * moved einsum_naive back to ops * remaining isn't needed * avoid creating another set * cleanup * greedy path, start of naive einsum * more einsum * fix some bugs * some more fixes, tests pass * benchmark * some simplify * fix einsum and test Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> * add a bunch more tests and fix a bunch more bugs * some docs nits --------- Co-authored-by: dc-dc-dc <dgcruz983@gmail.com> Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> |
||
|---|---|---|
| .. | ||
| mlx_tests.py | ||
| mpi_test_distributed.py | ||
| test_array.py | ||
| test_autograd.py | ||
| test_bf16.py | ||
| test_blas.py | ||
| test_compile.py | ||
| test_constants.py | ||
| test_conv.py | ||
| test_device.py | ||
| test_einsum.py | ||
| test_eval.py | ||
| test_fast_sdpa.py | ||
| test_fast.py | ||
| test_fft.py | ||
| test_graph.py | ||
| test_init.py | ||
| test_linalg.py | ||
| test_load.py | ||
| test_losses.py | ||
| test_metal.py | ||
| test_nn.py | ||
| test_ops.py | ||
| test_optimizers.py | ||
| test_quantized.py | ||
| test_random.py | ||
| test_reduce.py | ||
| test_tree.py | ||
| test_upsample.py | ||
| test_vmap.py | ||