TorchWI Full waveform inversion using PyTorch Install package python setup.py install Dependency pytorch horovod numpy, scipy, matplotlib