|
block.py
|
Run isort and black on python files
|
2023-08-18 14:22:11 -07:00 |
|
embedding.py
|
Run isort and black on python files
|
2023-08-18 14:22:11 -07:00 |
|
mha.py
|
[Gen] Use flash_attn_with_kvcache in generation
|
2023-09-07 08:24:43 -07:00 |
|
mlp.py
|
[MLP] Implement SwiGLU with torch jiterator
|
2023-09-04 15:43:53 -07:00 |
