Partitioned integrators for thermodynamic parameterization of neural networks