* Refactor npu mp to make it easier to integrate new models * fix style * move layer functions to base