Function precomputer_freqs_cis

Precompute the cos and sin of the frequencies for RoPE.

(
        dim: int,
        end: int = int(32 * 1024),
        rope_base: float = 1e6,
        rope_scaling: Optional[dict] = None,
)