If I understand the docs correctly, probabilities are expected as the model output, so it seems that applying a softmax should work.
However, this part of the docs:

the probabilities provided are assumed to be that of the positive class

sounds as if the loss function is dealing with a multi-label output (each output would give a probability in [0, 1] for a separate class)? Can you verify it?