Post #362 738 Sep 20, 2022, 16:10 UTC https://blog.tensorflow.org/2022/09/optimizing-tf-xla-and-jax-for-llm-training-on-nvidia-gpus.html@tensorflowblog blog.tensorflow.org Optimizing TF, XLA and JAX for LLM Training on NVIDIA GPUs Together, NVIDIA and Google are delighted to announce new milestones and plans to optimize TensorFlow and JAX for the Ampere and more.