FIX: debug mode for mlp layers