Identification of rheological parameters by physics in- formed neural networks. Application to granular media