mlx

MLX是苹果推出的在Apple silicon上运行的机器学习数组框架，提供类似NumPy和PyTorch的API，支持自动微分、延迟计算和动态图构建。