JAX

Python + Numpy 프로그램의 합성 변환

  • 소프트웨어 : JAX
  • OS : Linux,Windows
  • 프로그래밍 언어 : Python
  • 공식 웹사이트 : HOME

다운로드

JAX 상세

업데이트 된 AutoGrad 버전의 JAX는 자동으로 기본 파이썬 및 숫자 기능을 차별화 할 수 있습니다.루프, 가지, 재귀 및 폐쇄를 통해 차별화 될 수 있으며 파생 상품의 파생물의 파생물을 가져갈 수 있습니다.이는 Grad 및 전달 모드 차별화를 통해 역 모드 차별화 (A.K.A. Backpropagation)를 지원하며, 2 개는 임의로 임의로 구성 될 수 있습니다.새로운 기능은 JAX가 XLA를 사용하여 GPU 및 TPU에서 숫자 프로그램을 컴파일하고 실행하는 것입니다.컴파일은 기본적으로 후드에서 발생하며 라이브러리 호출이 Just-In-Time을 컴파일하고 실행하는 것입니다.그러나 JAX는 또한 하나의 기능 API 인 JIT를 사용하여 XLA 최적화 된 커널로 자신의 Python 함수를 자신의 Python 함수를 컴파일 할 수 있습니다.편집 및 자동 차별화는 임의로 구성 할 수 있으므로 정교한 알고리즘을 표현하고 Python을 남기지 않고 최대 성능을 얻을 수 있습니다.

Leave a Reply

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다