Comment Comment by kiratp Jax is the closest thing at the moment.https://developer.apple.com/metal/jax/And MLXhttps://github.com/ml-explore/mlx parent kiratp 2 years ago
https://developer.apple.com/metal/jax/
And MLX
https://github.com/ml-explore/mlx