JAX
Composable function transforms — the framework Gemini is built with.
आप एक software रिकॉर्ड देख रहे हैं: AI स्टैक का एक tool या platform — इसे कौन बनाए रखता है और यह किस hardware पर चलता है।
फ़ोटो: Google LLC · Public domain · Wikimedia Commons
JAX treats models as pure functions to be transformed — grad, jit, vmap, pmap — compiling through XLA to TPU pods with exceptional scaling ergonomics. It is Google DeepMind's house framework (Gemini training) and the favorite of much of the research frontier outside PyTorch.
यह बाकी स्टैक से कैसे जुड़ता है
इस रिकॉर्ड के लिंक एक नज़र में — किसी समूह पर क्लिक करके उस पर जाएँ।
पर चलता है
अनुरक्षक
प्रतिस्पर्धी
संबंधित
स्रोत
रिकॉर्ड की समीक्षा की गई 2026-01-15 — देखें पद्धति.