JAX新
我的评分
JAX 是 Google 开源的、NumPy 风格的数值计算库。它追踪 Python 函数并用 XLA 编译到 CPU/GPU/TPU，提供可组合的函数式变换——grad（自动微分）、jit（编译）、pmap/vmap（并行与向量化）。
对物理 AI 的价值：JAX 是现代物理 AI 的算力底座：JAX-CFD、JAX-Fluids 等可微求解器、神经算子，以及 PINN/神经 ODE 研究都构建在它「自动微分 + JIT」的基础之上。
标签:
JAX 是 Google 开源的、NumPy 风格的数值计算库。它追踪 Python 函数并用 XLA 编译到 CPU/GPU/TPU，提供可组合的函数式变换——grad（自动微分）、jit（编译）、pmap/vmap（并行与向量化）。
对物理 AI 的价值：JAX 是现代物理 AI 的算力底座：JAX-CFD、JAX-Fluids 等可微求解器、神经算子，以及 PINN/神经 ODE 研究都构建在它「自动微分 + JIT」的基础之上。
| / | Focus search |
| gh | Go to home |
| gf | Go to favorites |
| ga | Go to A–Z index |
| gr | Go to recent updates |
| gt | Go to timeline |
| gp | Go to Hot 100 |
| ? | Show this help |
| Alt+D | Toggle dark mode |
| Esc | Close dialog |