refactor autoencoder tests (temporal decoder, cosmos, kvae, mochi) (#13832)
* refactor autoencoder_kl_temporal_decoder tests
* refactor autoencoder_kl_cosmos tests
* refactor autoencoder_kl_kvae tests
* fix return_dict propagation in AutoencoderKLMochi.forward
* refactor autoencoder_kl_mochi tests
* add docstrings
* fix return type annotation
* remove unused base_precision and test_outputs_equivalence skip
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>