The MLX library released version 0.32.0, which addresses a specific bug in the Metal backend.

  • Fixes conv2 gradients for the grouped strided case on Metal.