- [ ] Shape handling should be done by the model. - [ ] Simple configuration in yaml - [ ] Downscaling - [ ] Reshaping - [ ] Dequantization - [ ] Remove any hardcoded downscaling