Learning 3D Granular Flow Simulations