jax-js (网站)

TLDR AI 工具
web-ml webgpu wasm javascript machine-learning compiler open-source

摘要

jax-js 是一个开源的机器学习库和 Web 编译器，使用 WebGPU 和 WebAssembly 内核将类似 JAX 的数值计算引入 JavaScript，全部在客户端运行。

jax-js 是一个机器学习库，可以运行神经网络、图像算法、模拟和数值代码，全部在浏览器中 JIT 编译。
查看原文
查看缓存全文

缓存时间: 2026/08/10 13:43

# jax-js – 开源 Web 机器学习框架 来源：https://jax-js.com/ ## jax-js 是一个面向 Web 的机器学习（ML）库和编译器 在 JavaScript 中提供高性能的 WebGPU 和 WebAssembly 内核。在你的浏览器中运行神经网络、图像算法、模拟和数值代码，全部通过 JIT 编译。 ## 将 jax-js 添加到你的项目 零依赖。支持并运行于所有主流浏览器。 > `npm install @jax-js/jax` ### 矩阵乘法 每秒数十亿次浮点运算（GFLOPs） Wasm WebGPU WebGPU-fp16 ## 在浏览器中，如同 JAX 和 PyTorch jax-js 是一个端到端的机器学习库，受 JAX 启发，但使用纯 JavaScript 实现： - 完全在客户端运行（Chrome、Firefox、iOS、Android）。 - 与 NumPy/JAX 具有紧密的 API 兼容性（https://github.com/ekzhang/jax-js/blob/main/FEATURES.md）。 - 从零编写，零外部依赖。 jax-js 很可能是最具可移植性的 GPU 机器学习框架，因为它可以在浏览器能运行的任何地方运行。它同样简单但经过优化，包含一个轻量级编译器，可将你的高层操作转换为 WebGPU 和 WebAssembly 内核。 jax-js 的目标是让数值代码对每个人都可访问、可部署，从而让计算密集型应用能在消费级硬件上快速本地运行。 ## 试试看！ 这是一个实时编辑器，代码正在你的浏览器中运行。 运行代码，在此处查看输出。 ## 在线示例 交互式演示，涵盖本地 AI 模型、模拟、几何处理和数值方法。 ## 资源

相似文章