Updates to `ReactingFlow` (and related) to support torch simulation target #380

Job	Run time
gpubaseimage / docker cuda base	11s
docker cpu env with cuda	9s
docker-gpu-arch-70 / docker gpu env	10s
docker-gpu-arch-80 / docker gpu env	6s
docker-gpu-arch-75 / docker gpu env	10s
	46s

Provide feedback