mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-16 09:55:33 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
broadcast_shape_2
tinygrad/extra/llama_kernels
History
wozeparrot e97f2c1114 llama: only gemm + fa custom kernel (#16180) 
* llama: tie store to grad directly

* llama: set mp flags

* llama: non fused grad fp8 quantize path
2026-05-12 21:03:49 -07:00
..
cast_amax
llama: only gemm + fa custom kernel (#16180)
2026-05-12 21:03:49 -07:00
fp8_transpose
llama speed 6 (#16071)
2026-05-06 20:51:03 -07:00
fused_ce
llama: speed 2 (#15960)
2026-04-28 20:44:37 -07:00
fused_pad_grad_accum
llama: speed 2 (#15960)
2026-04-28 20:44:37 -07:00
fused_rmsnorm_mul_quantize_fp8
llama mp fixes (#16050)
2026-05-05 15:35:32 -07:00
quantize_fp8_delayed
llama mp fixes (#16050)
2026-05-05 15:35:32 -07:00
rmsnorm
llama: move llama kernels to llama_kernels (#15952)
2026-04-27 22:48:53 -07:00
__init__.py
llama mp fixes (#16050)
2026-05-05 15:35:32 -07:00