Solved by albanD in post #4 
              
              
                This makes sense but I’ve seen a lot of code online that uses the non public API and if I’m not mistaken I’ve seen also some pytorch examples uses it as well. 

Can you link such example please? 

On another note, if you wanted to loop over the params of a module, do some operation and then remove …