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