lightning/docs/source-pytorch/accelerators/tpu.rst

.. _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>