User Guide:
User API:
Developer API:
Full API:
Functions
create_self_attention
Self attention block for Perceiver https://arxiv.org/pdf/2103.03206.pdf.