It depends. If you have TPUs, Jax. If you have GPUs, PyTorch. It also depends on who your community is, and whether you need to distribute models or not. For simple ML, either is fine and then it comes down to popularity within your circle.

→ View original post on X — @nandodf