Getting NaN in the softmax Layer

Is this only visible on the CPU, GPU or both devices?
Could you try to update numpy as suggested here?

CC @dhirajsuvarna