import numpy as np def softmax(x): max_x = np.max(x) exp_x = np.exp(x - max_x) return exp_x / np.sum(exp_x)