three\_body\_problem module =========================== See https://github.com/juntang-zhuang/TorchDiffEqPack/blob/master/test_code/three_body_problem.py