Revert "move weight to the front end"

This reverts commit 7999111667bca0d976c82a0a7b25e3926775ce60.