wozeparrot 0dc333cfab Promote Embedding to nn (#798) ... * feat: promote Embedding to nn * fix: fix failing test * feat: add test with jit * feat: rewrite embedding to no longer need stacked for loops * clean+fix: don't know how that happened		2023-05-25 18:39:45 -07:00
..
convnext.py	LayerNorm2d for 2 lines	2023-03-20 16:58:43 -07:00
efficientnet.py	third try at torch loading (#677)	2023-03-10 19:11:29 -08:00
resnet.py	getting 77% on imagenet eval	2023-05-13 07:46:27 -07:00
rnnt.py	Promote Embedding to nn (#798)	2023-05-25 18:39:45 -07:00
transformer.py	good changes from llama branch (#671)	2023-03-09 20:51:22 -08:00
unet3d.py	conv3d + unet3d (#772)	2023-05-12 13:54:07 -07:00
vit.py	good changes from llama branch (#671)	2023-03-09 20:51:22 -08:00