Tri Dao 48bc6eacd6 [Gen] Add rotary base as an argument to FT attention kernel		2023-05-30 13:38:34 -07:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
block.py	[BugFix] cannot unpack non-iterable NoneType object	2023-05-07 03:07:30 +09:00
embedding.py	Reorder LN in Block, support OPT	2023-01-15 22:14:31 -08:00
mha.py	[Gen] Add rotary base as an argument to FT attention kernel	2023-05-30 13:38:34 -07:00
mlp.py	Implement LLaMa	2023-04-18 21:51:35 -07:00