Add `allow_unused` arg for models with task-specific parameters

[rees-c]

Hi,

Thanks for building this library. I wanted to add the allow_unused argument when computing gradients to allow for meta-learners with task-specific parameters. I left the default behavior unchanged.