Для всех, кто начинает работать с глубоким обучением в 2025 году и хочет проводить обучение в больших масштабах: используйте JAX + Keras 3. Если только вам не нравится страдать.