RoPE base b: 10000 head dim d: 128