2 results found Sort:

lucidrains
block-recurrent-transformer-pytorch lucidrains
18
207
mit
8
Implementation of Block Recurrent Transformer - Pytorch
memory recurrence deep-learning attention-mechanisms long-context-attention artificial-intelligence long-context-transformers
Created 2023-02-07
62 commits to main branch, last one 11 months ago
lucidrains
flash-attention-jax lucidrains
23
182
mit
5
Implementation of Flash Attention in Jax
jax deep-learning attention-mechanisms long-context-attention artificial-intelligence
Created 2022-07-12
54 commits to main branch, last one 11 months ago