NOAH for Swin transformer??

I am trying to implement swin transformer with NOAH. Please provide some hints. I started with supernet_vision_transformer_timm.py file, but it became very complex to edit. 

I really appreciate any help you can provide. 