Add missing MochiEncoder3D.gradient_checkpointing attribute (#11146)
* Add missing 'gradient_checkpointing = False' attr
* Add (limited) tests for Mochi autoencoder
* Apply style fixes
* pass 'conv_cache' as arg instead of kwarg
---------
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>