Machine Learning Gravity Wave Parameterization Generalizes to Capture the QBO and Response to Increased CO2