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.

Installation

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.

Questions

Please contact alisk@rice.edu for questions.

Authors

Ali Siahkoohi and Mathias Louboutin