TorchTEM3D, a time-domain finite difference forward simulation platform for 3D transient electromagnetics. Can quickly calculate dBz/dt, and can quickly calculate sensitivity matrix based on automatic differentiation
Install
Create a new environment in Anaconda
conda create -n TorchTEM3D python=3.12
Install PyTorch
pip install torch==2.5.1+cu124 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
I have submitted to Computers&Geosciences. If you have any usage, please contact me：[email protected]
TorchTEM3D: PyTorch-Driven forward modeling platform for fast 3D transient electromagnetic modeling and efficient sensitivity matrix calculation https://doi.org/10.1016/j.cageo.2025.106063