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