Выносите ViewModel из Compose функций

Да, все верно, я использую LoginScreenRoot для того чтобы вынести методы viewModel.