test\_stiff module ================== See https://github.com/juntang-zhuang/TorchDiffEqPack/blob/master/test_code/test_stiff.py