JAX JAX 是由 Google 开源的 Python 数值计算与机器学习库,提供 NumPy 风格 API,并支持自动微分、即时编译、向量化和并行计算。其文档站点包含安装指南、核心概念、API 参考、教程与示例,适合研究人员和开发者了解 JAX 在高性能科学计算、深度学习模型构建和加速计算中的使用方法。 AI学习资源 # JAX # Python
NumPy NumPy 是 Python 生态中用于科学计算和数值分析的开源基础库,提供高性能的多维数组对象、向量化运算、线性代数、傅里叶变换、随机数等常用数值计算工具。它具备良好的兼容性和扩展性,可与 SciPy、pandas、Matplotlib、机器学习框架等工具协同使用,广泛应用于数据分析、工程计算、科研建模和人工智能开发等场景。 AI开发平台 # Python # 开源库