Дело вкуса. Фреймворки глубокого обучения не сильно отличаются друг от друга, у всех есть свои особенности. Для начинающих мог бы посоветовать использовать PyTorch - интуитивный и гибкий фреймворк, использование которого очень напоминает работу с NumPy.