Updates to memory registration and support for shared RX/TX contexts