tinygrad/extra/models
George Hotz b1c0d8c99d
remove cpu and torch backends (#3399)
* remove cpu and torch backends

* don't copy to cpu

* use clang instead of cpu

* multitensor gathers on the first device

* clang is cpu + use default

* fixup

* bugfix
2024-02-15 16:55:39 +01:00
..
bert.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
convnext.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
efficientnet.py remove cpu and torch backends (#3399) 2024-02-15 16:55:39 +01:00
llama.py sharding for llama (#3151) 2024-01-16 19:28:00 -08:00
mask_rcnn.py move dtypes to dtype.py (#2964) 2024-01-01 14:58:48 -08:00
resnet.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
retinanet.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
rnnt.py move graph.py and jit.py into features (#3376) 2024-02-12 17:34:34 +01:00
transformer.py replace with tensor op (#3099) 2024-01-12 14:13:40 -05:00
unet3d.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
vit.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00