Jacky LeeandGitHub 2e85fce068 Transformer: use Tensor.scaled_dot_product_attention (#1520)

2023-08-11 09:00:37 -07:00

..

bert.py

Corrected a few misspelled words (#1435)

2023-08-04 16:51:08 -07:00

convnext.py

LayerNorm2d for 2 lines

2023-03-20 16:58:43 -07:00

efficientnet.py

remove unused imports in models (#1088)

2023-07-01 12:04:19 -07:00

mask_rcnn.py

Add pylint trailing whitespace rule (#1314)

2023-07-21 13:37:55 -04: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 anys and alls (#1111)

2023-07-03 16:06:06 -07:00

rnnt.py

Promote Embedding to nn (#798)

2023-05-25 18:39:45 -07:00

transformer.py

Transformer: use Tensor.scaled_dot_product_attention (#1520)

2023-08-11 09:00:37 -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