Plumb through softmax scaling config to the builder.
Plumb through softmax scaling config to the builder.

PiperOrigin-RevId: 328668473

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo