Files
mlx/mlx/backend/metal/softmax.cpp