JAX | AI启航工具箱（OPEN AICHINES） - AI工具收录、分享与测试中心

什么是JAX?

JAX是由Google开源的一个用于机器学习和科学计算的Python库。它被设计成与NumPy非常相似的API接口，但具有额外的功能和优化，使其能够在CPU、GPU和TPU等硬件加速器上高效运行。

项目地址: https://github.com/google/jax

帮助文档: https://jax.readthedocs.io/en/latest/

快速入门链接：https://jax.readthedocs.io/en/latest/notebooks/quickstart.html

更多请看: 导航系统购买请至https://www.sslphp.com/ainv

相关导航

Lightning AI

Lightning AI官网,建立、培训和部署AI产品的平台The all-in-one platform for AI development. Code together. Prototype. Train. Scale. Serve. From your browser - with zero setup. From the creators of PyTorch Lightning.

caffe

caffe官网,深度学习流行的框架之一GitHub is where over 100 million developers shape the future of software, together. Contribute to the open source community, manage your Git repositories, review code like a pro, track bugs and features, power your CI/CD and DevOps workflows, and secure code before you commit it.

Leap

Leap官网,将AI快速集成到你自己的应用中

MindSpore

The all-scenario deep learning framework that best manifests the computing power of the Ascend AI processor, provides development experience with friendly MindSpore官网,华为ai自研的支持端边云全场景的深度学习训练推理框架昇思design and efficient execution for the data scientists and algorithmic engineers, and further advance the development and enrichment of the AI software/hardware application ecosystem.

fast ai

fast ai官网,基于PyTorch,简化深度学习的强大工具,优化模型训练与图像分类

DL4J

Introduction to core Deeplearning4j concepts.