HW-aware training extension to AIHWKIT-Lightning for MoEs - View it on GitHub