-record(router, {
    gate :: viva_tensor@tensor:tensor(),
    num_experts :: integer(),
    top_k :: integer(),
    noise_std :: float()
}).