lightning/docs/source/accelerators/tpu.rst

56 lines
1.2 KiB
ReStructuredText

.. _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: gpu_faq.html
:height: 150
.. raw:: html
</div>
</div>