I think an ultra wide band chip would need too much energy and would be too expensive.

A more reasonable choice in my mind would be to use the BLE 5.1 angle of arrival feature to find the direction of the head phones, as a ble module is already on board.

But therefore an antenna array would be necessary in the phone which I don’t think is the case, and I am not sure if manufacturers will do in the future.