Repolode
제출하기
patrick-kidger

patrick-kidger/diffrax

JAX 기반 수치 미분방정식 솔버로, 자동 미분과 GPU 실행을 지원하며 ODE, SDE, CDE를 처리합니다.

GitHub에서 보기
스타
2,122
포크
202
열린 이슈
265
마지막 커밋
2개월 전

어떤 프로젝트인가요?

Diffrax는 JAX 기반의 수치 미분방정식 솔버 라이브러리입니다. 일반 미분방정식(ODE), 확률 미분방정식(SDE), 제어 미분방정식(CDE)을 통합된 내부 구조로 처리하므로 세 종류의 방정식을 같은 방식으로 풀 수 있습니다.

라이브러리에는 명시적 적분을 위한 Tsit5, Dopri5, Dopri8을 비롯해 심플렉틱 및 암시적 방법 등 다양한 솔버가 포함되어 있습니다. 적분 구간을 포함한 모든 항목에 vmap을 적용할 수 있으며, 상태에는 임의의 PyTree를 사용할 수 있습니다. 밀집 해를 구할 수 있고, solve를 거친 역전파를 위한 여러 adjoint 방법도 제공됩니다.

Diffrax는 neural differential equation과 같이 미분 가능한 시뮬레이션이 필요한 사용자를 대상으로 합니다. pip으로 설치하며 Python 3.10 이상이 필요합니다.

핵심 기능

  • 01통합된 단일 구조로 ODE, SDE, CDE를 풉니다.
  • 02Tsit5와 Dopri8 등 명시적, 심플렉틱, 암시적 솔버를 포함합니다.
  • 03적분 구간을 포함해 solve 전반에 vmap을 적용할 수 있습니다.
  • 04PyTree 상태를 지원하고 밀집 해를 제공합니다.
  • 05역전파를 위한 여러 adjoint 방법을 제공합니다.

활용 사례

  • 솔버를 통과하는 그래디언트가 필요한 neural differential equation 모델을 학습합니다.
  • GPU에서 동역학 시스템의 배치 시뮬레이션을 실행합니다.
  • SDE 솔버로 확률 과정을 시뮬레이션합니다.

빠른 시작

  1. 1

    pip으로 라이브러리를 설치합니다

    pip install diffrax
  2. 2

    ODE 항과 솔버를 정의하고 diffeqsolve를 호출합니다(README의 간단한 예제 참조)

사용 전 알아둘 점

  • Python 3.10 이상이 필요합니다.
  • JAX를 기반으로 하므로 사용하려면 JAX가 제대로 설정되어 있어야 합니다.

포함된 컬렉션

GPU로 AI를 학습하고 실행하는 도구

토픽

#differential-equations#jax#scientific-computing#neural-differential-equations#ode-solver#gpu#deep-learning#dynamical-systems#equinox#machine-learning

자주 묻는 질문

diffrax은(는) 무엇인가요?
JAX 기반 수치 미분방정식 솔버로, 자동 미분과 GPU 실행을 지원하며 ODE, SDE, CDE를 처리합니다. Diffrax는 JAX 기반의 수치 미분방정식 솔버 라이브러리입니다. 일반 미분방정식(ODE), 확률 미분방정식(SDE), 제어 미분방정식(CDE)을 통합된 내부 구조로 처리하므로 세 종류의 방정식을 같은 방식으로 풀 수 있습니다.
diffrax은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
diffrax은(는) Apache-2.0 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
diffrax은(는) 아직 유지보수되고 있나요?
6개월 이내 업데이트. diffrax의 마지막 커밋은 2026년 9월 6일입니다.
diffrax은(는) 어디에 사용할 수 있나요?
솔버를 통과하는 그래디언트가 필요한 neural differential equation 모델을 학습합니다. GPU에서 동역학 시스템의 배치 시뮬레이션을 실행합니다. SDE 솔버로 확률 과정을 시뮬레이션합니다.
diffrax는 어떻게 설치하나요?
pip으로 라이브러리를 설치합니다: pip install diffrax ODE 항과 솔버를 정의하고 diffeqsolve를 호출합니다(README의 간단한 예제 참조)
diffrax은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/patrick-kidger/diffrax

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트