3D model

[CatEek]

Description

Adding 3D/2.5D to LVAE model

• What: Merge current implementation in the original repo with refactoring done previously.

Changes Made

• Relevant pydantic config
• Added encoder_conv_strides and decoder_conv_strides params. They control the strides in conv layers, and can have len 2 or 3. They're meant to make a choice between 2D and 3D convs and control the shape of encoder/decoder (subject to change)
• Input shape is now a tuple containing shapes of 2/3 dimensions
• Stochastic layer are in the separate module (subject to change)
• NonStochastic is removed alongside with relevant parameters
• Some docs update
• Basic tests

Notes/Problems

• Dosctings are a mess, should be fixed in a separate PR later
• Lot's of mypy etc issues
• Some tests don't pass because we need to clarify multiscale count param

---

Please ensure your PR meets the following requirements:

• [x ] Code builds and passes tests locally, including doctests
• [x ] New tests have been added (for bug fixes/features)
• [ ] Pre-commit passes
• [ ] PR to the documentation exists (for bug fixes / features)

[jdeschamps]

Can you update the description? Impossible to review otherwise. :)

[CatEek]

@federico-carrara Thanks for the review, updates some things, some were already fixed after you reviewed. I have a couple of questions though. Could you please take a look again