The authors of Two Facets of SDE Under an Information-Theoretic Lens: Generalization of SGD via Training Trajectories and via Terminal States have not publicly listed the code yet.
Request code directly from the authors:
Get an expert to implement this paper:
(OR if you have code to share with the community, please add it here 😊🙏 )