jax

Google에서 개발한 고성능 수치 계산 라이브러리로, 자동 미분과 GPU/TPU 가속을 지원한다. 머신러닝 연구 및 고성능 컴퓨팅 작업에서 널리 사용된다.