qazalandGitHub 7b6496f2e6 fix the reduceops cache breaking beautiful_mnist (#5938) ...

* fix the reduceops cache breaking beautiful_mnist

* test_sparse_categorical_crossentropy_simple

* starting tests

* atol from test_nn

* test_sparse_categorical_crossentropy_alt

* dont use torch

2024-08-07 00:02:54 +03:00

..

codegen

UOp generic div folding (#5896)

2024-08-05 21:38:43 -04:00

engine

fix the reduceops cache breaking beautiful_mnist (#5938)

2024-08-07 00:02:54 +03:00

nn

pow(2) -> square in RMSNorm [run_process_replay] (#5901)

2024-08-04 14:21:31 -04:00

renderer

add some types to the scheduler [run_process_replay] (#5941)

2024-08-06 12:23:54 -07:00

runtime

nv remove useless init (#5932)

2024-08-06 14:41:40 +03:00

shape

unify negative shape creation to raise ValueError (#5817)

2024-07-30 13:42:59 -04:00

__init__.py

add mnist tutorial

2024-04-25 16:08:32 +08:00

device.py

hcq copy queue is optional (#5909)

2024-08-05 14:03:25 +03:00

dtype.py

support using str to specify dtype (#5897)

2024-08-04 12:56:28 -04:00

function.py

threefry as pattern matcher (#5371)

2024-07-13 11:59:03 -07:00

helpers.py

fast mnist indexing (#5921)

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

lazy.py

fast mnist indexing (#5921)

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

multi.py

MLB support reshape for uneven shards (#5804)

2024-08-01 02:36:03 -07:00

ops.py

add test to validate lazyops dims (#5845)

2024-07-31 12:59:38 -07:00

tensor.py

move memory planner to realize (#5937)

2024-08-06 10:41:29 -07:00