  Partial differential equations (PDEs) are central to describing and modelling
complex physical systems that arise in many disciplines across science and
engineering. However, in many realistic applications PDE modelling provides an
incomplete description of the physics of interest. PDE-based machine learning
techniques are designed to address this limitation. In this approach, the PDE
is used as an inductive bias enabling the coupled model to rely on fundamental
physical laws while requiring less training data. The deployment of
high-performance simulations coupling PDEs and machine learning to complex
problems necessitates the composition of capabilities provided by machine
learning and PDE-based frameworks. We present a simple yet effective coupling
between the machine learning framework PyTorch and the PDE system Firedrake
that provides researchers, engineers and domain specialists with a high
productive way of specifying coupled models while only requiring trivial
changes to existing code.
