Reference Unofficial codebase for : [UNETR: Transformers for 3D Medical Image Segmentation], Ali Hatamizadeh, Dong Yang, Holger Roth, Daguang Xu. 2021. (https://arxiv.org/abs/2103.10504?context=cs.CV)