PyTorch、TensorFlow、JAX 等深度学习框架各有优劣，选型需结合实际场景。

参考链接

• https://zhuanlan.zhihu.com/p/61576496