JAX

machine learning frameworkko

Google에서 개발한 고성능 수치 계산 라이브러리로, 자동 미분 및 XLA 컴파일러를 활용하여 머신러닝 연구에 사용됩니다.