GitLab

* add habana

* add mask

* fix mask in outer_product_mean

* add dap

* add hmp

* merge training code

* add chunk for inference

* fix extra-msa stack for training

* support ddp in training

* fix inference bugs

* code refactoring for habana

* support hmp training

* enable all inference and train on Gaudi/Gaudi2 with optimized perf with latest base (#139)

* enable all inference and train on Gaudi/Gaudi2 with optimized perf

* refine code to adapt new base

* refine code to fix issues in code review
Co-authored-by: habanachina <habanachina@habana.ai>
Co-authored-by: Leo Zhao <48052473+LeoZhao-Habana@users.noreply.github.com>
Co-authored-by: habanachina <habanachina@habana.ai>