Files
mlx/mlx/backend/cuda/layer_norm.cu