It is a bug in MLX that has been fixed a few days ago:
https://github.com/ml-explore/mlx/pull/3083
I'm asking because it seems to have nice autodiff functionality. It even supports differentiating array mutation (https://ml-explore.github.io/mlx/build/html/usage/indexing.h...), which is something JAX and Zygote.jl can't do. Instead, both have ugly tricks like `array.at[index].set` and the `Buffer` struct.
So it would be cool to have this functionality on a "regular" CPU.
It is a modern and clean codebase without legacies, and I could understand most things without seeking external articles.
Is there a job/services that confirm that branding is appropriate across different languages? Seems like a non trivial problem to solve.