Multiple zero stage 3 related fixes (#3886)
* Option to override module apply
* Removing early partitioning in override
* Unit tests
* Cleanup
* Adapt unit test to succeed
* Handle missed params
* Add accelerate
* Code cleanup
* Add doc
* Add doc
* Add doc
