We have a User Pool set up in AWS Cognito for which we have also set up Azure AD as a SAML Identity provider for one of the app clients
The authentication flow works