JAX: High performance array computing — JAX documentation

7个月前发布 164 0 0

JAX: High performance array computing — JAX documentation

所在地:
加拿大
收录时间:
2025-09-20
JAX: High performance array computing — JAX  documentationJAX: High performance array computing — JAX documentation

JAX是一个用于高性能数组计算的开源库,可在CPU、GPU和TPU上运行。它结合了NumPy、Autograd和XLA的功能,支持自动微分、即时编译、自动向量化等,可帮助开发者高效地进行数值计算和机器学习开发。其重要性在于提升计算效率和开发灵活性,适用于科研、工业界的机器学习和深度学习项目。JAX是开源免费的,定位为高性能计算的编程工具。

数据统计

相关导航