For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Python module
max.experimental.nn.rope
Positional embedding modules and functions.
Rotary embeddings
RotaryEmbedding | Applies Rotary Positional Embeddings (RoPE) to input tensors. |
|---|---|
TransposedRotaryEmbedding | Applies RoPE using a transposed head-dimension layout. |