AMD related optimizations for transformer models - View it on GitHub