Techniques for controlling a bottom hole assembly (BHA) to follow a planned wellbore path include determining sensor measurements from the BHA; determining a model of BHA dynamics based on the sensor measurements from the BHA; determining a weighting factor that corresponds to a drilling objective; determining an objective function comprising the drilling objective, weighted by the weighting factor, and one or more constraints; determining a control input to the BHA that satisfies the objective function and the one or more constraints; and applying the control input to the BHA.