.. |
__init__.py
|
8e893f0950
Create __init__.py for ops/triton dir (#516)
|
1 year ago |
cross_entropy.py
|
22339db185
remove an unused import (#960)
|
6 months ago |
k_activations.py
|
f1a73d0740
Run isort and black on python files
|
1 year ago |
layer_norm.py
|
36587c01cb
[LayerNorm] Update layer_norm_linear
|
9 months ago |
linear.py
|
942fcbf046
[Rotary] Implement rotary in Triton
|
1 year ago |
mlp.py
|
f1a73d0740
Run isort and black on python files
|
1 year ago |
rotary.py
|
f692b98d80
Fix spurious re-compilations of `rotary_kernel` (#911)
|
8 months ago |