Separate prediction functionalities from inference

version 0.1.3 (Optimise GPU and memory)

version 0.1.2 (faster version by fixing torchdiffeq version)

Version 0.1.1

Modify doc badge