Note

From Ray 2.6.0 onwards, RLlib is adopting a new stack for training and model customization, gradually replacing the ModelV2 API and some convoluted parts of Policy API with the RLModule API. Click here for details.

Model APIs#

Base Model classes#

TorchModelV2

Feed Forward methods#

Recurrent Models API#

Acessing variables#

Customization#