Results 4 issues of acsweet

## Proposed changes These changes are an attempt to improve thread safety for the metal backend. This is related to #2067 Please let me know what you think. ## Checklist...

**Describe the bug** There seem to be some thread data access issues introduced in the newest version of mlx. **To Reproduce** This doesn't produce the exact same error I was...

**Describe the bug** There are noticeable differences between conv_general on gpu and cpu, and seems like overflow on the linux build. **To Reproduce** There are two code snippets, one with...

bug

## Description of the change This PR is the first (of more to come) to add support for the MLX backend of Keras to work with Keras Hub (see https://github.com/keras-team/keras/issues/19571)....