Waterflood management of production wells

A method of waterflood management for reservoir(s) having production hydrocarbon-containing well(s) including injector well(s). A reservoir model has model parameters in a mathematical relationship relating a water injection rate to a total production rate of the production well including at least one of a hydrocarbon production rate and water production rate. A solver implements automatic differentiation utilizing training data regarding the reservoir including operational data that includes recent sensor and/or historical data for the water injection rate and the hydrocarbon production rate, and constraints for the model parameters. The solver solves the reservoir model to identify values or value distributions for the model parameters to provide a trained reservoir model. The trained reservoir model uses water injection schedule(s) for the injector well to generate predictions for the total production rate.