mrq 6b0891448c pain (some shit to try and get some flash attention for ROCm (gfx1100) through triton fused attention but no good)		2024-08-25 20:07:27 -05:00
..
mamba_vasqu
retnet_syncdoth
__init__.py	pain (some shit to try and get some flash attention for ROCm (gfx1100) through triton fused attention but no good)	2024-08-25 20:07:27 -05:00
bitnet.py
llama.py	pain (some shit to try and get some flash attention for ROCm (gfx1100) through triton fused attention but no good)	2024-08-25 20:07:27 -05:00
mamba.py
mixtral.py
retnet.py
transformer.py