George HotzandGitHub 25847080f0 olmoe (from stream, wip) (#9390) ...

* olmoest working (but not)

* it's correct

* compare ropes

* old code wasn't wrong

* default device

* no metal

* fix permute

* working

* more minimal

2025-03-10 13:46:33 +08:00

..

bert.py

least_upper_float is at least default_float (#9303)

2025-02-28 10:41:56 -05:00

clip.py

…

convnext.py

…

efficientnet.py

…

inception.py

…

llama.py

olmoe (from stream, wip) (#9390)

2025-03-10 13:46:33 +08:00

mask_rcnn.py

explicitly check value for not None (#8382)

2024-12-23 11:12:39 -05:00

resnet.py

Fix FC layer ResNet load_from_pretrained error (#8387)

2024-12-26 18:11:27 -05:00

retinanet.py

combine pad2d with pad (#7677)

2024-11-14 17:56:02 +08:00

rnnt.py

…

t5.py

…

transformer.py

_one_hot_along_dim input needs to be int (#9179)

2025-02-20 09:00:43 -05:00

unet3d.py

…

unet.py

These casts should only happen if these are supported (#7644)

2024-11-12 07:56:50 +08:00

vit.py

…