JAX
8 posts
[JAX] 00. Introduction: JAX 실전 가이드
[JAX] 01. NumPy와 함수형 패러다임
[JAX] 02. jit과 XLA 컴파일
[JAX] 03. grad: 자동미분
[JAX] 04. vmap과 pmap: 벡터화와 병렬화
[JAX] 05. PRNG와 상태 관리
[JAX] 06. Flax·Optax로 신경망 실전
[JAX] 07. PyTorch에서 마이그레이션
Home