This bug is currently being used by Tanh, and Sigmoid and any other Linear subclass that makes use of _linear_part. I intend to fix this when I get your okay to start making changes to the mlp module.