google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
What it solves
JAXopt는 하드웨어 가속, 배치 가능성, 미분 가능성을 갖춘 옵티마이저 라이브러리입니다. 사용자는 GPU와 TPU를 사용하여 최적화 문제를 해결하고, 이 문제들의 솔루션을 입력값에 대해 미분할 수 있습니다.
How it works
JAX 기반으로 구축된 이 라이브러리는 JAX의 vmap을 활용하여 자동 벡터화(automatic vectorization)를 수행함으로써 동일한 최적화 문제의 여러 인스턴스를 동시에 처리합니다. 최적화 솔루션의 미분은 암시적 미분(implicit differentiation) 또는 언롤링된 알고리즘 반복(unrolled algorithm iterations)에 대한 autodiff를 사용하여 지원합니다.
Who it’s for
미분 가능 프로그래밍 워크플로우에 통합할 수 있는 고성능 옵티마이저 도구가 필요한 JAX 사용자 연구원 및 개발자입니다.
Highlights
CPU, GPU, 및 TPU를 위한 하드웨어 가속.
JAX의
vmap을 통한 최적화 문제의 배치를 위한 자동 벡터화.최적화 솔링션의 암시적 미분 및 언롤링된 미분 모두 지원.
관련
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트