ATurker

Optimization Plans for Conv2D CPU Execution

3

Hello, I wonder if there are any future plans to optimize Conv2D CPU execution. I guess currently MLX uses a naive implementation?

fix: conv_general differences between gpu, cpu

3

## Proposed changes Fixes inconsistencies in conv_general results between GPU and CPU implementations by properly handling asymmetric padding in the CPU backend. Implements separate padding_lo and padding_hi parameters throughout the...

Raise Error When Setting requires_grad=True on Non-Float Tensors

7

Closes #9723. This PR ensures that tensors with non-floating point data types cannot have requires_grad=True.