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