MLX 是 Apple 机器学习研究团队推出的数组计算框架，面向 Apple Silicon 优化，支持惰性求值、统一内存、函数变换、编译、分布式通信和 LLM 推理示例。

JAX 是面向高性能数组计算和机器学习研究的开源 Python 库，提供自动微分、JIT 编译、自动向量化、并行化、PyTree、随机数和分布式数组能力。