SeisBench: A toolbox for machine learning in seismology

SeisBench is an open-source Python toolbox for machine learning in seismology. It brings together the whole machine learning model lifecycle: datasets and benchmarks, models and training pipelines, and efficient implementations for deploying the models in production.


Cite this software

What SeisBench: A toolbox for machine learning in seismology can do for you

PyPI - License GitHub Workflow Status Read the Docs PyPI Python 3.9 DOI

The Seismology Benchmark collection (SeisBench) is an open-source python toolbox for machine learning in seismology. It provides a unified API for accessing seismic datasets and both training and applying machine learning algorithms to seismic data. SeisBench has been built to reduce the overhead when applying or developing machine learning techniques for seismological tasks.

Getting started

SeisBench offers three core modules, data, models, and generate. data provides access to benchmark datasets and offers functionality for loading datasets. models offers a collection of machine learning models for seismology. You can easily create models, load pretrained models or train models on any dataset. generate contains tools for building data generation pipelines. They bridge the gap between data and models.

The easiest way of getting started is through our colab notebooks.

Dataset basicsOpen In Colab
Model APIOpen In Colab
Generator PipelinesOpen In Colab
Applied pickingOpen In Colab
Using DeepDenoiserOpen In Colab
Depth phases and earthquake depthOpen In Colab
Training PhaseNet (advanced)Open In Colab
Creating a dataset (advanced)Open In Colab
Building an event catalog (advanced)Open In Colab

Alternatively, you can clone the repository and run the same examples locally.

For more detailed information on Seisbench check out the SeisBench documentation.


SeisBench can be installed in two ways. In both cases, you might consider installing SeisBench in a virtual environment, for example using conda.

The recommended way is installation through pip. Simply run:

pip install seisbench

SeisBench is build on pytorch. As of pytorch 1.13.0, pytorch is by default shipped with CUDA dependencies which increases the size of the installation considerably. If you want to install a pure CPU version, the easiest workaround for now is to use:

pip install torch==1.12.1 seisbench

We are working on a more permanent solution that allows to use the latest pytorch version in a pure CPU context.

Alternatively, you can install the latest version from source. For this approach, clone the repository, switch to the repository root and run:

pip install .

which will install SeisBench in your current python environment.


There are many ways to contribute to SeisBench and we are always looking forward to your contributions. Check out the contribution guidelines for details on how to contribute.

Known issues

  • We've recently changed the URL of the SeisBench repository. To use the new URL update to SeisBench 0.4.1. It this is not possible, you can use the following commands within your runtime to update the URL manually:
    import seisbench
    from urllib.parse import urljoin
    seisbench.remote_root = ""
    seisbench.remote_data_root = urljoin(seisbench.remote_root, "datasets/")
    seisbench.remote_model_root = urljoin(seisbench.remote_root, "models/v3/")
  • On the Apple M1 and M2 chips, pytorch seems to not always work when installed directly within pip install seisbench. As a workaround, follow the instructions at ( to install pytorch and then install SeisBench as usual through pip.
  • Some institutions and internet providers are blocking access to our data and model repository, as it is running on a non-standard port (2880). This usually manifests in timeouts when trying to download data or model weights. To verify the issue, try accessing directly from the same machine. We are working on a permanent solution for the issue. In the meantime, if you are having trouble, try downloading through another network/VPN if possible. You can also contact your network administrator to allow access to port 2880 on our server. Otherwise, reach out to us, and we will work on finding a solution.
  • EQTransformer model weights "original" in version 1 and 2 are incompatible with SeisBench >=0.2.3. Simply use from_pretrained("original", version="3") or from_pretrained("original", update=True). The weights will not differ in their predictions.


Reference publications for SeisBench:


The initial version of SeisBench has been developed at GFZ Potsdam and KIT with funding from Helmholtz AI. The SeisBench repository is hosted by HIFIS - Helmholtz Federated IT Services.

Logo of SeisBench: A toolbox for machine learning in seismology
Programming languages
  • Jupyter Notebook 69%
  • Python 31%
  • GPL-3.0-only
</>Source code

Participating organisations

Helmholtz Centre Potsdam GFZ German Research Centre for Geosciences
Karlsruhe Institute of Technology (KIT)
Helmholtz Centre For Ocean Research Kiel (GEOMAR)
Helmholtz AI