A Pedagogical Framework for Physics-Informed Machine Learning: From Classical Pendulum to Quantum Anharmonic Oscillator Using PyTorch on Modern GPU Hardware
Abstract
We present a five-module pedagogical framework for teaching physics-informed machine learning (ML) through two progressively complex physical systems: a driven, damped nonlinear pendulum and a one-dimensional quantum anharmonic oscillator. Five model architectures are implemented and compared: a standard artificial neural network (ANN), a one-dimensional convolutional neural network (CNN), a long short-term memory (LSTM) network, and two physics-informed neural networks (PINNs) -- one per physical system. All models are implemented in PyTorch~2.9 and executed on an NVIDIA RTX~5090 GPU, making the framework directly applicable to modern deep learning laboratory courses. Quantitative benchmarks show that data-driven models achieve mean absolute errors of ~rad (pendulum ANN) and ~a.u.\ (quantum CNN), while the curriculum-trained pendulum PINN reaches an MAE of ~rad using only collocation points. A systematic CPU-vs-GPU benchmark reveals speedups ranging from (small ANN) to (LSTM), providing a concrete pedagogical demonstration of when GPU acceleration is -- and is not -- warranted. The framework is packaged as self-contained Jupyter notebooks designed for a graduate-level \emph{Deep Neural Networks for Physical Systems} course, with embedded reflection questions that guide students from data-driven thinking toward physics-constrained formulations.
Cite
@article{arxiv.2502.05621,
title = {A Pedagogical Framework for Physics-Informed Machine Learning: From Classical Pendulum to Quantum Anharmonic Oscillator Using PyTorch on Modern GPU Hardware},
author = {Enis Yazici},
journal= {arXiv preprint arXiv:2502.05621},
year = {2026}
}
Comments
11 pages, 9 figures