George Hotz 791530045d Refactor LoadOps (#910) ... * test * work * upd test * loadops * cleanups * real ones * remove LazyNumpyArray * fix assign test * remove range * np.require * llama uses arange kernels * no caching consts * fix enet * torch load support * tests cleanup * fix shufflenet * fix image * fix torch_load test		2023-06-03 09:40:43 -07:00
..
bert.py	Add mlperf bert model (#803)	2023-05-27 14:53:32 -07:00
convnext.py	LayerNorm2d for 2 lines	2023-03-20 16:58:43 -07:00
efficientnet.py	Refactor LoadOps (#910)	2023-06-03 09:40:43 -07:00
resnet.py	add retinanet with resnet backbone (#813)	2023-05-28 20:20:16 -07:00
retinanet.py	add retinanet with resnet backbone (#813)	2023-05-28 20:20:16 -07:00
rnnt.py	Promote Embedding to `nn` (#798)	2023-05-25 18:39:45 -07:00
transformer.py	default transformer dropout to 0 (#828)	2023-05-29 08:06:16 -07:00
unet3d.py	Add MLPerf UNet3D model (#775)	2023-05-28 20:38:19 -07:00
vit.py	good changes from llama branch (#671)	2023-03-09 20:51:22 -08:00