Skip to main content
Ctrl+K
JAX-KR 문서 - Home

Getting Started

  • 🔪JAX - 세부적인 특징들 (JAX - The Sharp Bits)🔪
  • JAX 101
    • 가속화된 NumPy인 JAX
    • JAX의 Just-In-Time 컴파일
    • JAX의 자동 벡터화
    • JAX의 고급 자동 미분
    • JAX의 유사난수생성(PRN)
    • 파이트리 (Pytrees) 사용하기
    • JAX에서 병렬 평가(Parallel Evaluation)
    • JAX에서 상태를 유지하는 연산
  • Flax Tutorial
    • Quick Start
    • Flax 기초
    • 파라미터와 상태 관리하기
    • Flax 모듈 인자 다루기 (Dealing with Flax Module arguments)
    • 전체 데이터셋 처리
    • 배치 정규화 (Batch normalization)
  • JAX-backend-Keras
    • 케라스 코어 시작하기 (Getting started with Keras Core)
    • JAX로 fit()에서 일어나는 일 사용자 정의하기 (Customizing what happens in fit() with JAX)
    • JAX로 스크래치부터 학습 루프 만들어보기 (Writing a training loop from scratch in JAX)
    • JAX로 다중-GPU 분산학습 진행하기 (Multi-GPU distributed training with JAX)
  • .rst

JAX-backend-Keras

JAX-backend-Keras#

Keras Core

  • 케라스 코어 시작하기 (Getting started with Keras Core)
  • JAX로 fit()에서 일어나는 일 사용자 정의하기 (Customizing what happens in fit() with JAX)
  • JAX로 스크래치부터 학습 루프 만들어보기 (Writing a training loop from scratch in JAX)
  • JAX로 다중-GPU 분산학습 진행하기 (Multi-GPU distributed training with JAX)

이전

배치 정규화 (Batch normalization)

다음

케라스 코어 시작하기 (Getting started with Keras Core)

으로 JAX/Flax Lab

© Copyright 2023, JAX/Flax Lab.