That’s a good decision. I know both Keras and PyTorch, and I will recommend PyTorch any day. I haven’t used TF/Keras in the last 2.5 years outside of Edge AI

That’s a good decision. I know both Keras and PyTorch, and I will recommend PyTorch any day. I haven’t used TF/Keras in the last 2.5 years outside of Edge AI projects- neither at work nor at personal projects. JAX is interesting, though. But definitely not advisable as someone's first DL framework.