Hi,
Could you try to disable the “Media style notification” in AudioRelay’s output settings? (Other topic on the subject)
If the issue “music disconnects when locking the phone” is still there:
Please try to disable all the battery optimization for both Plex Amp and AudioRelay as indicated in Samsung | Don’t kill my app!