mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
616e9c1483* work * work * the assembly * remove the old one * remove ws bufs, assert splitk * notes cleanup * work * gemm args * gemm in mixins would be nice * add gemm gradient * print counters * the realize is for DEBUG=2 aesthetics * dedup * rewrite to python dsl, no list copies * leave that * add B, M, N, K to gemm name * it's M0 not NULL * fp16 support * test cleanup + more gemms * work from viz * more work * gemm batch_size * xccg path work * tiny comments on the label naming * s_waitcnt |
||
|---|---|---|
| .. | ||
| scripts | ||
| training_submission_v4.0/tinycorp | ||
| training_submission_v4.1/tinycorp | ||
| training_submission_v5.0/tinycorp | ||
| training_submission_v5.1/tinycorp | ||
| training_submission_v6.0/tinycorp | ||
| dataloader.py | ||
| helpers.py | ||
| initializers.py | ||
| losses.py | ||
| lr_schedulers.py | ||
| metrics.py | ||
| model_eval.py | ||
| model_spec.py | ||
| model_train.py | ||
| README | ||
Each model should be a clean single file. They are imported from the top level `models` directory It should be capable of loading weights from the reference imp. We will focus on these 5 models: # Resnet50-v1.5 (classic) -- 8.2 GOPS/input # Retinanet # 3D UNET (upconvs) # RNNT # BERT-large (transformer) They are used in both the training and inference benchmark: https://mlcommons.org/en/training-normal-21/ https://mlcommons.org/en/inference-edge-30/ And we will submit to both. NOTE: we are Edge since we don't have ECC RAM