JAX

JAX는 구글 리서치가 만든 고성능 수치 연산 라이브러리로, NumPy 같은 API에 자동 미분과 JIT 컴파일을 더했어요. GPU·TPU에서 즉시 실행을 지원하고, 함수를 그대로 변환해 그라디언트·벡터화·병렬화를 손쉽게 얻을 수 있죠. 미니멀한 함수형 설계 덕분에 연구 코드를 빠르고 확장성 있게 만드는 데 강점이 있어요.