I have the AirPods Pro’s and usually don’t use them on the subway. I tend to use my Sony WH-1000XM4 and notice the sound better since it was released. I know they support 360° audio but didn’t think Apple Music was supported.
If it has the "Dolby Atmos" logo while you're playing the song, it's active. You can go into Settings and have Spatial Audio/Dolby Atmos either set to Automatic (to only be active while an Apple-approved capable listening device is being used) or "Always On", so it always plays that format despite how you're listening (which might mean messy audio on some headphones).