jax-ml/oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
해결하는 문제
Oryx는 확률적 프로그래밍 및 딥러닝을 위한 프레임워크를 제공하여 사용자가 JAX의 고성능 컴퓨팅 기능과 통합될 수 있는 복잡한 통계 모델을 구축할 수 있도록 합니다.
작동 방식
jit(저스트 인 타임 컴파일), grad(자동 미분), vmap(벡터화)과 같은 JAX의 핵심 변환과 원활하게 구성되고 통합되도록 설계된 일련의 함수 변환을 구현합니다.
대상
확률적 모델링과 JAX의 속도 및 유연성을 결합해야 하는 연구원 및 개발자.
주요 기능
- JAX 위에 구축됨
- 확률적 프로그래밍 지원
jit,grad,vmap과 같은 JAX 변환과 통합
관련
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트