Terramechanics models augmented by machine learning representations. (June 2023)