JAX JAX 是由 Google 开源的 Python 数值计算与机器学习库,提供 NumPy 风格 API,并支持自动微分、即时编译、向量化和并行计算。其文档站点包含安装指南、核心概念、API 参考、教程与示例,适合研究人员和开发者了解 JAX 在高性能科学计算、深度学习模型构建和加速计算中的使用方法。 AI学习资源 # JAX # Python
MLX MLX 是面向 Apple Silicon 的机器学习与数值计算框架,由 Apple 机器学习研究团队开发。其提供类似 NumPy 的数组接口、自动微分、惰性计算、模型训练与分布式计算能力,支持在 Mac 等 Apple 芯片设备上高效运行和开发人工智能应用。 AI学习资源 # Apple Silicon # 分布式计算