| .. |
|
triton
|
bedf877467
[CrossEntropy] Fix where labels address not aligned to 16 bytes
|
hai 2 meses |
|
__init__.py
|
ece539abd6
Add __init__.py files to subdirectories for installation
|
%!s(int64=2) %!d(string=hai) anos |
|
activations.py
|
7153673c1a
Fix swiglu backwards return type (#1337)
|
hai 4 semanas |
|
fused_dense.py
|
0cb595ad94
[bugfix] handle_x not define when using checkpoint_lvl = 2 (#502)
|
hai 1 ano |
|
layer_norm.py
|
f1a73d0740
Run isort and black on python files
|
hai 1 ano |
|
rms_norm.py
|
f1a73d0740
Run isort and black on python files
|
hai 1 ano |
Neil Tenenholtz
7153673c1a
Fix swiglu backwards return type (#1337)