mlx.nn.ALiBi

class ALiBi

Methods

create_alibi_matrix(q_sequence_length, ...)
create_alibi_slope(num_heads)