FEATURE] Optimize kernel representations for Triton

[LoserCheems]

Summary

• Introduces optimized kernel representation functions for forward and backward operations.

Root Cause

• The existing kernel functions lacked optimized representations, affecting performance.

Changes

• Added new kernel representation functions and refactored existing kernel functions to utilize these optimized representations.

Reproduction

• Not applicable as this is a feature enhancement.

Tests

• Validated performance improvements through existing benchmarks.

Compatibility

• No backward compatibility issues.