Softmax

Use amdlibm for the exp in softmax when available.