George Hotz b14d31d6db ConvNeXt + extras (#657) ...

* simple convnext implementation

* shorter function names

* need to realize the random functions now

* creating an optimizer realizes all params

* assign contiguous

* fix lazy lazy

* why was i doing that...add convnext to tests

* LazyNumpyArray

* enable assert + comment

* no two tiny

2023-03-06 22:10:56 -08:00

..

convnext.py

ConvNeXt + extras (#657)

2023-03-06 22:10:56 -08:00

efficientnet.py

BatchNorm2D -> BatchNorm2d (#558)

2023-02-16 12:31:49 -08:00

resnet.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

transformer.py

rename log_softmax, support dim, fix onnx Softmax

2023-02-24 10:11:24 -08:00

vit.py

support larger ViT models

2021-12-12 10:45:10 -08:00