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

相关导航

Leap

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

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.

DL4J

Introduction to core Deeplearning4j concepts.

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.

NumPy

NumPy官网,Python语言的一个扩展程序库

fast ai

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