[gemini] support amp o3 for gemini

[ver217]

Overview

Current AMP of gemini only supports O2, and O3 (no master weights) is not supported. We should support this by adding an argument master_weights of gemini.