MLX 是 Apple 机器学习研究团队推出的数组计算框架,面向 Apple Silicon 优化,支持惰性求值、统一内存、函数变换、编译、分布式通信和 LLM 推理示例。
JAX 是面向高性能数组计算和机器学习研究的开源 Python 库,提供自动微分、JIT 编译、自动向量化、并行化、PyTree、随机数和分布式数组能力。
Caffe 是 BAIR 和社区贡献者开发的开源深度学习框架,以表达性、速度和模块化为目标,曾广泛用于视觉、语音、多媒体研究与工业部署。