George HotzandGitHub b4bf6a7dea switch backward to use gradient [pr] (#8235) ...

* switch backward to use gradient [pr]

* set device correctly, dedup

* why does that fail?

* add noop cast

* simple backward

* fix beautiful_mnist

* touchups

* set in compute_gradient

* uop_count

* uop_count was wrong

* collections

* no note

* skip that test

* update sched kernel counts

* train mnist is 65

* fix metadata and gc

* fixes

* materialize_grads

* no pathlib stuff

* add contiguous_backward, fix bugs

* add some realize

* fix multi

2025-01-26 09:12:16 +09:00

..

efficientnet

test cleanups

2023-03-03 06:36:06 -08:00

waifu2x

waifu2x vgg7: testcase, auto-RGBA->RGB, function to grab pretrained models, training "fix" (#2117)

2023-10-19 22:07:15 -07:00

whisper

whisper: support batch inference, add librispeech WER test (#2074)

2023-11-16 13:50:08 -08:00

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_bert.py

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_efficientnet.py

db in wal mode (#5388)

2024-07-12 20:43:36 -07:00

test_end2end.py

tests from grad uop path [pr] (#8313)

2024-12-18 09:25:05 -08:00

test_mnist.py

skip a few clang test that took > 30 seconds in CI (#4126)

2024-04-10 02:00:34 -04:00

test_onnx.py

make onnx runner a class (#8647)

2025-01-20 10:11:05 -08:00

test_real_world.py

switch backward to use gradient [pr] (#8235)

2025-01-26 09:12:16 +09:00

test_resnet.py

Fix FC layer ResNet load_from_pretrained error (#8387)

2024-12-26 18:11:27 -05:00

test_rnnt.py

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

test_train.py

Remove pytest markers (#2831)

2023-12-18 18:53:28 -05:00

test_waifu2x.py

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_whisper.py

move is_dtype_supported to device [pr] (#7575)

2024-11-07 20:38:03 +08:00