mlx.nn.ALiBi#

class ALiBi#

Methods

create_alibi_matrix(q_sequence_length, ...)

create_alibi_slope(num_heads)