serverless\_gpu package
=======================

Submodules
----------

serverless\_gpu.compute module
------------------------------

.. automodule:: serverless_gpu.compute
   :members: GPUType
   :show-inheritance:


serverless\_gpu.launcher module
-------------------------------

.. automodule:: serverless_gpu.launcher
   :members: distributed, DistributedFunction
   :show-inheritance:


serverless\_gpu.runtime module
------------------------------

.. automodule:: serverless_gpu.runtime
   :members: get_local_rank, get_local_world_size, get_node_rank, get_world_size, get_global_rank, get_replica_rank
   :show-inheritance:


serverless\_gpu.ray module
---------------------------

.. automodule:: serverless_gpu.ray
   :members: ray_launch
   :show-inheritance: