Para quem está a começar com deep learning em 2025 e procura fazer treino em grande escala: use JAX + Keras 3. A menos que goste de sofrer.