Tri Dao bedf877467 [CrossEntropy] Fix where labels address not aligned to 16 bytes 2 月之前
..
triton bedf877467 [CrossEntropy] Fix where labels address not aligned to 16 bytes 2 月之前
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation 2 年之前
activations.py 3557e0bb8f [MLP] Implement SwiGLU with torch jiterator 1 年之前
fused_dense.py 0cb595ad94 [bugfix] handle_x not define when using checkpoint_lvl = 2 (#502) 1 年之前
layer_norm.py f1a73d0740 Run isort and black on python files 1 年之前
rms_norm.py f1a73d0740 Run isort and black on python files 1 年之前