David Hou 9a485f36e4 shard kvcache (#5830)		2024-07-30 20:29:54 -07:00
..
bert.py	Residual in MLM loss + Change default steps (#4935)	2024-06-12 16:09:18 -04:00
clip.py	added clip vision model (#5595)	2024-07-19 18:35:51 -04:00
convnext.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
efficientnet.py	simple LoadOps.ASSIGN (#3745)	2024-03-14 20:44:34 -07:00
inception.py	FID Inception Model (#5516)	2024-07-16 23:12:03 -04:00
llama.py	shard kvcache (#5830)	2024-07-30 20:29:54 -07:00
mask_rcnn.py	remove numpy from dtype (#4969)	2024-06-14 15:38:45 -04:00
resnet.py	Fix batchnorm shapes for resnet.load_pretrained (#5167)	2024-06-26 18:44:10 -04:00
retinanet.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
rnnt.py	change Tensor.stack to method (#4719)	2024-05-24 17:04:19 -04:00
transformer.py	replace with tensor op (#3099)	2024-01-12 14:13:40 -05:00
unet.py	Pulled CLIP and UNet into Seperate Files (#5253)	2024-07-01 22:33:01 -04:00
unet3d.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00
vit.py	move to new cached fetch (#2493)	2023-11-28 17:36:55 -08:00