Text this: A machine learning-based simplified collision model for granular flows.