JAX(라이브러리)

딥 러닝 라이브러리

[ 주요 라이브러리 ]

JAX

버전	0.7.0 2025년 7월 23일 업데이트
공개일	2020년 5월 8일
\|

1. 개요

1. 개요 [편집]

JAX는 고성능 수치 계산과 대규모 머신러닝을 위해 설계된, 가속기[1] 친화적인 배열[2] 계산 및 프로그램 변환을 위한 Python 라이브러리입니다.
공식 깃헙 레포 소개글의 첫 문장[원문]

JAX (Just Another XLA)는 구글이 만든 신경망 기반 기계학습 프레임워크로 PyTorch의 장점을 반영하고 TensorFlow를 대체하고자 하는 목적으로 개발되었다.

[1] 이 문맥에서는 GPU, TPU 등 대량의 계산을 병렬적으로 빠르게 처리하기 위해 만들어진 컴퓨터 부품을 뜻한다.[2] 행렬(수학), 쉽게 말하자면 숫자들의 묶음이라고 생각해도 좋다.[원문] JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.

이 문서의 내용 중 전체 또는 일부는

JAX(라이브러리)

문서의 r6 판에서 가져왔습니다. 이전 역사 보러 가기