.. include:: /_includes/rllib/we_are_hiring.rst

.. _rllib-distributions-reference-docs:

Distribution API

.. include:: /_includes/rllib/new_api_stack.rst

.. currentmodule:: ray.rllib.models.distributions

Base Distribution class

.. autosummary:: :nosignatures: :toctree: doc/

~Distribution
~Distribution.from_logits
~Distribution.sample
~Distribution.rsample
~Distribution.logp
~Distribution.kl