tinygrad/extra/llama_kernels/cast_amax
2026-05-28 04:33:07 +09:00
..
__init__.py llama: don't allocate grad_xw13 in bf16 (#16359) 2026-05-28 04:33:07 +09:00
cast_amax_bwd_w13.cpp llama: don't allocate grad_xw13 in bf16 (#16359) 2026-05-28 04:33:07 +09:00
cast_amax_fwd_w13.cpp llama: speed 2 (#15960) 2026-04-28 20:44:37 -07:00