tahamajs/RR / Docs /module_metadata /fused_svd_kernel.json
tahamajs's picture
download
raw
685 Bytes
{
"module_name": "fused_svd_kernel",
"class_name": "FusedSVDLinear",
"category": "attention",
"family": "kernel",
"params": [
"in_features",
"out_features",
"rank",
"bias",
"use_rms_norm",
"eps",
"dtype",
"device"
],
"ops": [
"linear",
"view",
"reshape"
],
"docstring": "RESEARCH SOTA: Fused SVD-Linear Module.\nCombines low-rank projection $Z = x V^T$, in-SRAM RMSNorm, and up-projection $Y = Z U^T + b$\ninto a single kernel invocation, eliminating HBM roundtrips.",
"rank": 32,
"d_state": 1,
"num_experts": 1,
"top_k": 1,
"sparsity_pct": 0.0,
"bits": 4,
"hash_val": 14373005152563282559452131682605764698
}

Xet Storage Details

Size:
685 Bytes
·
Xet hash:
5c847a8785aa392aaeeae7329870be13f1f9b13fc4cc388ab9a48121d042beea

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.