I could be wrong, I have never used keras but for what I know there is not a precise mapping between keras (that has a high-level abstraction) and pytorch.
The thing most similar to keras.Model, I think it is nn.Module but I am not sure.
Let’s wait for someone else more expert than me