maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

Published 20 hours ago • verified publisher icon google

Add support for local sliding window attention in TPU splash_attention

Open gagika opened this issue 6 months ago • 0 comments

Added support for sliding window attention masking in TPU splash_attention

gagika avatar Aug 19 '24 05:08 gagika