(Python edit-able) implementation of the GRU, not in lower-level languages?

Hello, thanks in advance.

I have been doing research on the LSTM using the edit-able implementation located at:
/python3.10/site-packages/torch/nn/quantizable/rnn.py

I am now trying to do the same for the GRU, however I cannot find a similar edit-able implementation.
Is there an edit-able version of the GRU in Pytorch?

By “edit-able” I mean that I can edit it in Python code, as I don’t know the lower level languages