/neurodiffeq

A light-weight & flexible library for solving differential equations using neural networks based on PyTorch. Spatial-temporal ODEs and PDEs are both supported.

Primary LanguagePythonMIT LicenseMIT

Watchers