Machine-learning-based models in particle-in-cell codes for advanced physics extensions