对比 PyTorch、TensorFlow、JAX、Theano，我发现都在关注两大问题

OneFlow

2022-12-21

最近，我在处理 PyTorch 分布式和 TorchRec 相关的工作，为此，我开始学习 PyTorch 2.0。在业余时间，我也在跟着Alpa作者学习JAX和XLA。如今回顾这些技术，我发现它们的关注点似乎都是两个问题。