mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
* cleanup llama apply_rotary_emb and other helpers used ellipsis and other higher level tensor function. disabled the half @ half -> half tensor core as it fails uop dtype checks * keep hip 8x8->8 wmma |
||
|---|---|---|
| .. | ||
| bert.py | ||
| convnext.py | ||
| efficientnet.py | ||
| llama.py | ||
| mask_rcnn.py | ||
| resnet.py | ||
| retinanet.py | ||
| rnnt.py | ||
| transformer.py | ||
| unet3d.py | ||
| vit.py | ||