compute-qkv

compute-qkv state weights layer-idx configprocedure: Computes q, k and v from xb.