Raw code for Gated Recurrent Unit (GRU) and/or Optimized GRU

I’m looking to modify the OGRU (Wang et al, 2019) for multivariate time series prediction. Is there anyone who has stable code for the OGRU or at least a GRU built from scratch(not calling the Python package)?