raven-ml/raven

해결하는 문제

Raven은 OCaml 언어를 위한 수치 계산, 머신러닝, 데이터 과학을 위한 포괄적인 라이브러리 생태계를 제공합니다. Python의 과학 기술 스택(NumPy, JAX, PyTorch 등)의 기능을 OCaml로 가져오되, 언어의 강력한 타입 안전성을 유지하는 것을 목표로 합니다.

작동 방식

Raven은 ML 파이프라인의 다양한 부분을 처리하는 전문 패키지로 구성되어 있습니다:

• nx: 다차원 배열과 선형 대수를 다룹니다.
• tolk: GPU 텐서 계산을 위한 최소한의 ML 컴파일러입니다.
• rune: 자동 미분과 함수형 변환을 제공합니다.
• rune과 kaun: 함께 사용하여 신경망을 구축하고 훈련합니다.
• brot: 빠르고 HuggingFace 호환 토크나이저를 구현합니다.
• talon: 타입 안전한 데이터프레임을 제공합니다.
• hugin: 발표 수준의 플로팅을 위해 사용됩니다.
• quill: 탐색을 위한 인터랙티브한 REPL과 마크다운 노트북입니다.
• munin: TUI 대시보드를 갖춘 로컬 실험 추적입니다.
• sowilo: 미분 가능한 컴퓨터 비전입니다.
• fehu: 강화 학습 환경입니다.

대상 사용자

OCaml에서 과학적 계산과 머신러닝을 수행하고 싶은 개발자 및 연구자. 타입 안전성과 함수형 프로그래밍 패러다임의 이점을 누릴 수 있습니다.

주요 특징

• 풀스택 ML 생태계: Python의 생태계와 마찬가지로 데이터프레임(Talon), 토크나이제이션(Brot) 등을 포함합니다.
• 타입 안전한 연산: OCaml의 타입 시스템으로 과학적 계산 도구를 재구성합니다.
• GPU 가속: Tolk를 통해 GPU 텐서 계산을 위한 컴파일러를 포함합니다.
• 자동 미분: Rune을 통해 통합된 자동 미분입니다.
• 통합: HuggingFace 호환 토크나이제이션과 RL 환경을 제공합니다.