JAX JAX 是由 Google 开源的 Python 数值计算与机器学习库,提供 NumPy 风格 API,并支持自动微分、即时编译、向量化和并行计算。其文档站点包含安装指南、核心概念、API 参考、教程与示例,适合研究人员和开发者了解 JAX 在高性能科学计算、深度学习模型构建和加速计算中的使用方法。 AI学习资源 # JAX # Python