[JAX] 00. Introduction: JAX 실전 가이드
![[JAX] 00. Introduction: JAX 실전 가이드](/post/jax/getting-started-jax/wordcloud_hu_4af74fab2e13b766.webp)
Collections
8 pages
[JAX] JAX 실전 가이드
NumPy 호환 배열과 함수형 순수성부터 jit·XLA 컴파일, grad 자동미분, vmap·pmap 벡터화·병렬화, PRNG 상태 관리, Flax·Optax 신경망 구축, PyTorch 마이그레이션까지 다루는 8편 시리즈입니다.
[JAX] 01. NumPy와 함수형 패러다임
![[JAX] 01. NumPy와 함수형 패러다임](/post/jax/numpy-and-functional-paradigm/wordcloud_hu_d00a081c12d1d353.webp)
[JAX] 02. jit과 XLA 컴파일
![[JAX] 02. jit과 XLA 컴파일](/post/jax/jit-and-xla-compilation/wordcloud_hu_d6d19df3d2846872.webp)
[JAX] 03. grad: 자동미분
![[JAX] 03. grad: 자동미분](/post/jax/autodiff-with-grad/wordcloud_hu_f6123283e7e09f89.webp)
[JAX] 04. vmap과 pmap: 벡터화와 병렬화
![[JAX] 04. vmap과 pmap: 벡터화와 병렬화](/post/jax/vmap-and-pmap/wordcloud_hu_e8ef4cc8674ecb21.webp)
[JAX] 05. PRNG와 상태 관리
![[JAX] 05. PRNG와 상태 관리](/post/jax/prng-and-state-management/wordcloud_hu_93681f8782cd9130.webp)
[JAX] 06. Flax·Optax로 신경망 실전
![[JAX] 06. Flax·Optax로 신경망 실전](/post/jax/neural-networks-with-flax-optax/wordcloud_hu_3785a02ae9ac88e1.webp)
[JAX] 07. PyTorch에서 마이그레이션
![[JAX] 07. PyTorch에서 마이그레이션](/post/jax/migrating-from-pytorch/wordcloud_hu_7c05820792806d79.webp)