Yes, It is supported. We have detailed documentation about moving from ADFS to PTA/PHS as that is the scenario most of the customers fall under.
You can follow the standard deployment guide for setting up a federation in Azure AD for this migration.
Ref: https://learn.microsoft.com/en-us/azure/active-directory/hybrid/how-to-connect-fed-whatis