AMD Layers
==========

High-level layer abstractions for AMD GPUs.

Layer List
----------

.. list-table:: Available AMD Layers
   :header-rows: 1
   :widths: 30 70

   * - Layer
     - Description
   * - :doc:`tp_attn`
     - Tensor Parallel Attention layer
   * - :doc:`tp_mlp`
     - Tensor Parallel MLP layer

.. toctree::
   :maxdepth: 1
   :hidden:

   tp_attn
   tp_mlp