The actual code represents the work performed by Sofia Strukova during Google Summer of Code (GSoC) 2021, as part of the ML4SCI organization. It was performed under the supervision of Patrick Peplowski (JHUAPL), Sergei Gleyzer (University of Alabama) and Jason Terry (University of Georgia).
The code of the work described hereby can be found in the following open-source repository: link