Groundwater flow equation: Finite-difference forward, adjoint, and gradient operators with PyTorch wrapper
This repository contains the code for implementing the finite-difference forward, adjoint, and gradient operators for the groundwater flow equation. The operators are implemented via Devito, a finite-difference domain-specific language for solving partial differential equations that generates optimized C code depending on the target architecture.
Clone the repository and install the package in editable mode.
# Create a new conda environment (optional).
conda create --name groundwater python
conda activate groundwater
# Clone the repository and install the package in your Python environment.
git clone ttps://github.com/alisiahkoohi/groundwater
cd groundwater/
pip install -e .
Visit the Devito installation guide for more information on setting up the environment variables to fully utilize the parallelization capabilities of Devito.
Please contact alisk@rice.edu for questions.
Ali Siahkoohi and Mathias Louboutin