hasktorch/hasktorch

해결하는 문제

Hasktorch는 Haskell 프로그래밍 언어를 사용하여 텐서와 신경망을 구축할 수 있는 방법을 제공하며, 강력한 타입 안전성과 함수형 환경에서 PyTorch의 핵심 C++ 라이브러리의 능력을 활용할 수 있게 해줍니다.

작동 방식

이 라이브러리는 PyTorch가 공유하는 핵심 C++ 라이브러리(libtorch)를 활용하는 Haskell 라이브러리입니다. 저수준 외부 함수 인터페이스(FFI)와 코드 생성을 사용하여 PyTorch의 C++ 기능을 Haskell에 바인딩합니다. 라이브러리는 빌드 과정 중에 CPU, CUDA, Apple의 MPS 백엔드를 포함한 다양한 플랫폼용 공식 libtorch 바이너리를 자동으로 다운로드하고 캐시합니다.

대상 사용자

함수형 프로그래밍 언어를 사용하여 딥러닝 모델과 텐서 연산을 구현하고자 하는 Haskell 개발자 및 연구자.

주요 특징

• PyTorch 통합: PyTorch가 사용하는 동일한 C++ 코어 위에 구축됨.
• 크로스플랫폼 지원: Linux(CPU/CUDA), macOS Apple Silicon(CPU/MPS), macOS Intel(CPU)에서 작동.
• 자동 설정: 빌드 과정 중에 필요한 libtorch 바이너리를 자동으로 다운로드하고 캐시.
• 유연한 빌드 도구: Cabal, Stack, Nix와 호환.