tinygrad/test/unit/test_function.py at 5ff278446cf4a413be5ae0e66ea1097072cbbe43

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00

Files

George Hotz bb84e389cf functions for llama trainer (#15045 )

* functions for llama trainer

* function there

* axis match

* fix multi

* lil cleaner

* there's a bug with HK_FLASH_ATTENTION

* training functions

* for commit

2026-02-28 12:15:18 +08:00

11 KiB

Raw Blame History

View Raw

11 KiB Raw Blame History

11 KiB

Raw Blame History