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