feat: enhance attention visualization with token context and improve markdown explanations

fix: made the sequence length accurate for real model setup and also added information on which attention head is attending which set of dims

feat: add explanation on `hidden_size = num_attention_heads * dim_per_head`

feat: increase seq_len and auto-update of dim on UI during compute

feat: add rope visualization only with some minor cmparison with sinusoidial positional embedding