Learning MRI contrast agnostic registration