add masked multi-stream loss module