Array API 2024 tracking issue

[tomwhite]

This issue tracks the work needed to support Python array API standard v2024.12.

The changes from the previous version are listed here: https://data-apis.org/array-api/2024.12/changelog.html#v2024-12

New array functions

elemwise

These should be straightforward to add.

• nextafter
• reciprocal

indexing

• take_along_axis

searching

• count_nonzero

statistical

• cumulative_prod

utility

• diff

Updates

• Add max dimensions to capabilities
• Add support for complex input to mean
• Add support for scalars in result_type
  • We delegate result_type to the backend array library so this shouldn't need any changes.
• Add scalar support to where
• Add scalar support to elemwise functions