How to use state embedding in RLlib

Hi,
I’m getting started with Ray RLlib and I’m interested in embedding my states to a higher dimensional space so that the model can take different input sizes.
Is there an example/tutorial how to do that with Ray RLlib?
Thanks