George Hotz 718ced296c move state to nn/state (#1619)		2023-08-22 07:36:24 -07:00
..
bert.py	Bert: use Tensor.scaled_dot_product_attention (#1528)	2023-08-12 08:46:04 -07:00
convnext.py	LayerNorm2d for 2 lines	2023-03-20 16:58:43 -07:00
efficientnet.py	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
mask_rcnn.py	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
resnet.py	remove unused imports in models (#1088)	2023-07-01 12:04:19 -07:00
retinanet.py	Use generators instead of lists in `any`s and `all`s (#1111)	2023-07-03 16:06:06 -07:00
rnnt.py	np.argmax to Tensor.argmax (#1608)	2023-08-21 15:22:29 -07:00
transformer.py	.cpu().numpy() -> .numpy() (#1594)	2023-08-21 09:53:29 -07:00
unet3d.py	Add pylint trailing whitespace rule (#1314)	2023-07-21 13:37:55 -04:00
vit.py	good changes from llama branch (#671)	2023-03-09 20:51:22 -08:00