Modelling an ARCH process

[mingruimingrui]

An AR processes can be modelled with tfp.sts.Autoregressive.
How would one add a conditional heteroskedasticity component to a model?

If I receive a hint, I can perhaps make a notebook for future reference!