I would recommend to take a look at this tutorial 2 to get familiar with writing custom nn.Modules in PyTorch.
Based on the posted Keras model you could define these layers in the __init__ with minor changes (change e.g. Convolution2D to nn.Conv2d) and use these layers in the forward method.