API Reference
=============

API reference for launching distributed GPU workloads from Databricks notebooks.

distributed
-----------

.. currentmodule:: serverless_gpu

.. autofunction:: distributed

The decorator returns a :class:`~serverless_gpu.launcher.DistributedFunction` object.
Call ``.distributed()`` on it, with the same arguments you would pass to your original
function, to launch execution on GPU.

GPU Types
---------

.. currentmodule:: serverless_gpu.compute

.. note::

   ``gpu_type`` is auto-detected from the notebook's attached GPU. If you do specify it,
   you can directly pass ``"H100"`` or ``"A10"`` as a string.

.. autoclass:: GPUType
   :show-inheritance:
   :exclude-members: H100, A10, values