Tri Dao 41a21d6204 Fix import error ...

Fix #1618

2025-04-25 14:48:27 -04:00

..

__init__.py

Add __init__.py files to subdirectories for installation

2022-11-17 16:55:44 -08:00

block.py

[LayerNorm] Switch from CUDA to Triton implementation

2024-01-05 00:31:17 -08:00

embedding.py

Run isort and black on python files

2023-08-18 14:22:11 -07:00

mha.py

Fix import error

2025-04-25 14:48:27 -04:00

mlp.py

Add Alibi to MHA, test with Baichuan-13B

2023-12-21 22:49:55 -08:00