JAX
소프트웨어
학습
Active
Composable function transforms — the framework Gemini is built with.
현재 보고 계신 것은 소프트웨어 레코드입니다: AI 스택의 도구 또는 플랫폼 — 누가 관리하며 어떤 하드웨어에서 실행되는지를 다룹니다.
사진: 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 — 보기 방법론.