lancerts 22339db185 remove an unused import (#960) 6 months ago
..
__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