mlx.nn.log_softmax# mlx.nn.log_softmax(x, axis=- 1)# Applies the Log Softmax function. Applies \(x + \log \sum_i e^{x_i}\) element wise.
