Back to Pytorch Lightning

Accelerator: TPU training

docs/source-pytorch/accelerators/tpu.rst

2.6.41.2 KB
Original Source

.. _tpu:

Accelerator: TPU training

.. raw:: html

<div class="display-card-container">
    <div class="row">

.. Add callout items below this line

.. displayitem:: :header: Prepare your code (Optional) :description: Prepare your code to run on any hardware :col_css: col-md-4 :button_link: accelerator_prepare.html :height: 150 :tag: basic

.. displayitem:: :header: Basic :description: Learn the basics of single and multi-TPU core training. :col_css: col-md-4 :button_link: tpu_basic.html :height: 150 :tag: basic

.. displayitem:: :header: Intermediate :description: Scale massive models using cloud TPUs. :col_css: col-md-4 :button_link: tpu_intermediate.html :height: 150 :tag: intermediate

.. displayitem:: :header: Advanced :description: Dive into XLA and advanced techniques to optimize TPU-powered models. :col_css: col-md-4 :button_link: tpu_advanced.html :height: 150 :tag: advanced

.. displayitem:: :header: FAQ :description: Frequently asked questions about TPU training. :col_css: col-md-4 :button_link: tpu_faq.html :height: 150

.. raw:: html

    </div>
</div>