That would be very nice… I would send it as a feature request to support@airtable.com.
One thing to note is that the record doesn’t actually move to the new location until you’re done editing the record. So it does give you time to stay in the record and make your updates. Of course, it’s not very helpful when the record suddenly disappears from you as soon as you’re done editing.