torch
seisbench
torchinfo
matplotlib
numpy
ipykernel