Refactor: Remove unnecessary type casting in softmax#338

erdenebayrd wants to merge 1 commit intometa-llama:mainmeta-llama/llama3:mainfrom

erdenebayrd:refactorerdenebayrd/llama3:refactorCopy head branch name to clipboard