RoPEΒΆ

Rotary Positional Embeddings.

Abstract Signature:

RoPE(dims: int)

PyTorch

API: β€”
Strategy: Custom / Partial

Apple MLX

API: mlx.nn.RoPE
Strategy: Direct Mapping