Tutorial: JAX 101

Tutorial

• 가속화된 NumPy인 JAX
• JAX의 Just-In-Time 컴파일
• JAX의 자동 벡터화
• 수동 벡터화
• 자동 벡터화
• 결합된 변환
• JAX의 고급 자동 미분
• JAX의 유사난수생성(PRN)
• 파이트리 (Pytrees) 사용하기
• JAX에서 병렬 평가(Parallel Evaluation)
• JAX에서 상태를 유지하는 연산