Allow arrays in MTKParameters pullback

[DhairyaLGandhi]

Checklist

• Appropriate tests were added
• Any code changes were done in a way that does not break public API
• All documentation related to code changes were updated
• The new code follows the
  contributor guidelines, in particular the SciML Style Guide and
  COLPRAC.
• Any new documentation only uses public API

Additional context

Allows handling arrays in the case that we get arrays in the pullback for MTKParameters.

Add any other context about the problem here.