perf(xla): pre-transpose the LM head for single-sequence decode - #2864

perf(xla): pre-transpose the LM head for single-sequence decode#2864

inureyes wants to merge 1 commit into

mainlablup/mlxcel:mainfrom

perf/2863-xla-head-layoutlablup/mlxcel:perf/2863-xla-head-layoutCopy head branch name to clipboard