2 results found Sort:

lucidrains
block-recurrent-transformer-pytorch lucidrains
19
214
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
65 commits to main branch, last one 2 months ago
lucidrains
flash-attention-jax lucidrains
23
194
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 about a year ago