tensorflow - 简单来说，JAX、Trax 和 TensorRT 之间有什么区别？

Question

我一直在使用 TensorRT 和 TensorFlow-TRT 来加速我的 DL 算法的推理。
然后我听说过:

JAX https://github.com/google/jax

特拉克斯 https://github.com/google/trax

两者似乎都加速了深度学习。但我很难理解他们。谁能用简单的术语解释它们？

Answer 1

Trax是谷歌创建并被谷歌大脑团队广泛使用的深度学习框架。它是 TensorFlow 的替代品和 PyTorch当谈到实现现成的最先进的深度学习模型时，例如 Transformers、Bert 等，原则上与自然语言处理领域有关。Trax建立在 TensorFlow和 JAX . JAX是 Numpy 的增强和优化版本。关于JAX的重要区别和 NumPy是前者使用名为 XLA(高级线性代数)的库，它允许运行您的 NumPy代码在 GPU和 TPU而不是在 CPU就像它发生在平原 NumPy ，从而加速计算。