ML-based physics modeling