Hi @rijobro , I just found the `AutoEncoder` and `VarAutoEncoder` classes does not have docstrings, should we add? Thanks!