thermal mpinn.py