jax-ml/jax 源码下载_部署教程

项目概述

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

项目地址

https://github.com/jax-ml/jax

项目页面预览

jax-ml/jax preview

关键指标

Stars：34586
主要语言：Python
License：Apache License 2.0
最近更新：2026-01-15T12:15:13Z
默认分支：main

本站高速下载（国内可用）

点击下载（本站镜像）
– SHA256：7ef59241596387e505ef056fd08c49bd93d6d900220dee9fcf69b3d1edb70b65

安装部署要点（README 精选）

Installation

常用命令（从 README 提取）

@software{jax2018github,
  author = {James Bradbury and Roy Frostig and Peter Hawkins and Matthew James Johnson and Chris Leary and Dougal Maclaurin and George Necula and Adam Paszke and Jake Vander{P}las and Skye Wanderman-{M}ilne and Qiao Zhang},
  title = {{JAX}: composable transformations of {P}ython+{N}um{P}y programs},
  url = {http://github.com/jax-ml/jax},
  version = {0.3.13},
  year = {2018},
}