나무모에 미러 (일반/밝은 화면)
최근 수정 시각 : 2026-05-09 16:58:26

Flax



머신러닝 라이브러리
Accord.NET · Flax · JAX · Keras · MaxText · ML.NET · MLX · NumPy · PyTorch · TensorFlow · XLA

{{{#!wiki style="margin:-10px -10px"<tablebordercolor=#356fa5> 파일:파이썬_화이트.svgPython
패키지 목록
}}}
{{{#!wiki style="margin: 0 -10px -5px; min-height: 28px;"
{{{#!folding [ 펼치기 · 접기 ]
{{{#!wiki style="margin: -6px -1px -11px; word-break: keep-all;"
<colbgcolor=#356fa5><colcolor=#fff> 수학 NumPy
데이터 가공 Pandas, Polars, PySparkPandas on Spark 포함
데이터 시각화 matplotlib, Plotly
웹 프레임워크
동기
Django, Flask
비동기
FastAPI
머신 러닝·딥 러닝 TensorFlow · Keras, PyTorch, JAX · Flax
프론트엔드·GUI PyQt, wxPython, PyGTK
게임 Pygame, Ren'Py
기타 SQLAlchemyDB, OpenCV컴퓨터 비전, RequestsHTTP, OpenZL데이터 압축, PlaywrightCUA
}}}}}}}}} ||
1. 개요2. 관련 문서

1. 개요

Flax는 JAX 기반의 신경망 라이브러리로, 구글 브레인 팀이 개발하였다. 함수형 프로그래밍 스타일을 기반으로 하며, 고수준 딥러닝 모델 작성과 학습을 위한 모듈과 API를 제공한다. XLA를 활용한 최적화, TPU 및 GPU 분산 학습을 지원하며 연구 환경에서 널리 사용된다.

2. 관련 문서