Differential Equations as a Pytorch Neural Network Layer

How to use differential equations layers in pytorch

Kevin Hannay
Towards Data Science
17 min readApr 26, 2023

--

Differential equations are the mathematical foundation for most of modern science. They describe the state of a system using an equation for the rate of change (differential). It is remarkable how many systems can be well described by equations of this form. For example, the physical laws describing motion, electromagnetism and quantum mechanics all take this form. More broadly, differential equations describe chemical reaction rates through the law…

--

--