Files
thermal-conduction-ml/ThermalSolver/model/learnable_finite_difference.py