Model optimization custom layers support

Hello TF friends,

I saw this blog post Adding Quantization-aware Training and Pruning to the TensorFlow Model Garden — The TensorFlow Blog related with CustomLayer support etc for QAT.
Though no mention of version of packages (like TF/ TF MOT etc) involved is mentioned.
Can anyone help…