klyn.math.nn.attention.GroupedQueryAttention.GroupedQueryAttention
constructor
public GroupedQueryAttention(heads as Long, kvHeads as Long, width as Long, capacity as Long, rotary as Long, epsilon as Double = 1e-6)
Description

Allocates sequence-owned caches. No past tensor is recopied when appending.

Parameters
ParameterDescription
headsQuery head count.
kvHeadsKV head count dividing heads.
widthHead dimension.
capacityMaximum token count for this sequence.
rotaryEven rotated width, at most width.
epsilonPositive RMS stabilizer.
Throws
  • ValueException for invalid dimensions.
Example

state = GroupedQueryAttention(16l, 4l, 256l, 2048l, 64l)