patrick-kidger/optimistix

해결하는 문제

Optimistix는 비선형 문제(근 찾기, 최소화, 고정점, 최소제곱 등)를 통합적이고 모듈식 프레임워크로 해결하는 라이브러리입니다. JAX 생태계를 활용하여 이러한 다양한 수학적 문제를 변환하고 해결하는 과정을 단순화합니다.

작동 방식

JAX 기반으로 구축되어 자동 미분, 자동 병렬화, GPU/TPU 지원을 활용합니다. PyTrees를 사용하여 상태를 관리하며 Optax와의 상호 운용성을 제공합니다. 사용자는 예를 들어 BFGS 2차 그릇과 도그레그 강하 경로, 신뢰 영역 업데이트를 조합한 모듈식 최적화기를 정의할 수 있으며, 문제 유형 간 자동 변환도 가능합니다. 예를 들어 근 찾기 문제를 최소제곱 문제로 변환할 수 있습니다.

대상 사용자

과학 계산 및 머신러닝 분야에서 작업하는 연구자 및 개발자에게 적합합니다. 고성능이고 미분 가능한 비선형 소거기를 필요로 하는 분들께 추천합니다.

주요 특징

• 상호 운용성 있는 소거기: 문제 유형 간 자동 변환 (예: 근 찾기 → 최소제곱).
• 모듈식 최적화기: 강하 경로와 업데이트 규칙의 사용자 정의 조합 가능.
• JAX 네이티브: 자동 미분, GPU/TPU 가속, 빠른 컴파일링 완전 지원.
• Optax 통합: DeepMind의 Optax 라이브러리와 원활하게 연동.