Jax-JS：JavaScript实现的高性能WebGPU机器学习库

Jax-JS是一款纯JavaScript实现的机器学习库，专为WebGPU优化，可在浏览器中高效运行类似Google JAX的ML任务。它利用WebAssembly和WebGPU技术，支持自动微分、内核融合等高级功能，实现近原生性能（如500 GFLOP/s）。示例包括MNIST神经网络训练和CLIP模型实时语义搜索，性能超越TensorFlow.js。项目已开源，旨在将强大ML框架引入前端，推动AI应用在客户端的普及。