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 anys and alls (#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