The MLX library released version 0.32.1, introducing support for a head dimension of 72 within its Metal full attention implementation.

  • The update enables the use of head dimension 72 in Metal full attention via pull request #4330.