Lint

Use shared log_softmax kernels from PyTorch #43311

Job	Run time
android-java-format / linux-job	4m 50s
lintrunner / linux-job	12m 0s
link-check / lint-urls / linux-job	4m 50s
link-check / lint-xrefs / linux-job	4m 51s
	26m 31s

Provide feedback