‏jax-js: یادگیری ماشین کامل توی مرورگر

jax-js یه کتابخونهٔ یادگیری ماشین و کامپایلر برای وبه. تو معرفی پروژه اومده که کاملاً از صفر و با جاوااسکریپت خالص نوشته شده و هیچ وابستگی بیرونی نداره. کاری که می‌کنه اینه که شبکه‌های عصبی، الگوریتم‌های تصویری، شبیه‌سازی‌ها و کد عددی رو مستقیم توی مرورگر اجرا می‌کنه، اون هم به‌صورت JIT کامپایل‌شده.

قلب ماجرا یه کامپایلر سبکه که عملیات سطح‌بالا رو به کرنل‌های WebGPU و WebAssembly تبدیل می‌کنه. به گفتهٔ سازنده، API این کتابخونه سازگاری نزدیکی با NumPy و JAX داره، پس اگه با اون‌ها کار کرده باشین محیط براتون آشناست. روی همهٔ مرورگرهای اصلی هم کار می‌کنه: کروم، فایرفاکس، iOS و اندروید.

نویسنده معتقده همین موضوع jax-js رو احتمالاً قابل‌حمل‌ترین فریم‌ورک یادگیری ماشین مبتنی بر GPU می‌کنه، چون هرجا مرورگر اجرا بشه این هم اجرا می‌شه. سایت پروژه یه REPL زنده و مجموعه‌ای از دموهای تعاملی داره: چت با یه مدل زبانی کوچیک، رونویسی صدا با Whisper، تبدیل متن به گفتار، شبیه‌سازی سیال، تشخیص اشیا با ONNX و آموزش شبکه روی MNIST.

هدفی که برای پروژه اعلام شده اینه که کد عددی برای همه در دسترس و قابل‌عرضه بشه تا اپ‌های محاسبات‌سنگین سریع و محلی روی سخت‌افزار معمولی کاربر اجرا بشن. یه بنچمارک ضرب ماتریسی هم روی خود سایت هست تا سرعتش با بقیهٔ کتابخونه‌های وب مقایسه بشه. نصبش هم با پکیج @jax-js/jax روی npm انجام می‌شه.

نکات کلیدی:

• کتابخونه و کامپایلر یادگیری ماشین برای وب، با جاوااسکریپت خالص و بدون هیچ وابستگی
• کامپایل JIT عملیات به کرنل‌های WebGPU و WebAssembly
• سازگاری نزدیک API با NumPy و JAX
• اجرای کامل سمت کلاینت روی کروم، فایرفاکس، iOS و اندروید
• دموهای زنده: چت مدل زبانی، Whisper، TTS، شبیه‌سازی سیال و آموزش MNIST