Physics-driven machine learning models coupling PyTorch and Firedrake | Arena Library | Arena