Reminds me of a post on here in 2018, "It Takes Two Neurons to Ride a Bicycle (2004)" [1], where many of the commenters rightfully pointed out that instead of turning this into a machine learning problem, you could just implement a PID controller [2] to solve the problem. With control theory you can model a dynamic process like this more directly, and the theories have been around for a long time already.
ML shouldn't be applied where it's not necessary.
[1] https://news.ycombinator.com/item?id=16215130
[2] https://en.wikipedia.org/wiki/PID_controller