Saved toolsSubmit a tool
Models & infrastructure
TOOL PROFILE

JAX

Google推出的用于变换数值函数的机器学习框架

Models & infrastructureAI开发平台
Translation is queued. This page updates when ready.
Claim this toolVisit websiteCompare toolsFeedback & corrections
JAX

About this tool

JAX是Google推出的高性能数值计算库，提供类似NumPy的API，支持GPU/TPU加速、自动微分、即时编译（JIT）和向量化等功能。

JAX is a Python library for accelerator-oriented array computation and program transformation, designed for…

JAX provides a familiar NumPy-style API for ease of adoption by researchers and engineers.

Key features

  • 自动微分
  • 即时编译（JIT）
  • 向量化
  • 并行化
  • 硬件加速
  • 程序变换

Use cases and audience

待编辑核验

Input待编辑核验
→
Output待编辑核验

How to use

  1. 01

    环境配置与安装

  2. 02

    创建Python环境

  3. 03

    安装JAX库

  4. 04

    使用JAX的主要功能

Sources

JAX: High performance array computing — JAX documentationGoogle JAX - Google推出的用于变换数值函数的机器学习框架 | AI工具集Updated on 2026-10-17

Before you start

Check availability, registration, limits and plans on the official site. Try a small sample before choosing.

This profile uses public official information, without full testing. Commercial terms, privacy, regional access and limits need further checks.

Basic profile · More checks pending

Scope: name, official link and basic purpose.

View original source

You may also like

View all