luminal-ai/luminal

Inference at the speed of light.

해결하는 문제

Luminal은 전통적인 딥러닝 프레임워크에서 발생하는 성능 저하와 복잡성 폭증을 극복하기 위해 설계되었습니다. 즉각 실행(eager-first) 방식을 대체하여, 사전에 계산 그래프를 최적화하는 고성능의 일반 목적 추론 컴파일러를 도입함으로써 런타임 오버헤드를 줄이고 하드웨어 활용도를 극대화합니다.

작동 방식

Luminal은 모든 신경망을 정적 계산 그래프로 간주합니다. 복잡한 연산을 15개의 기본 연산(unary, binary, 기타)으로 축소하여 최소한의 핵심을 유지합니다. 수작업 히ュ리스틱에 의존하는 대신, 탐색 기반 접근법을 사용하여 가장 효율적인 실행 경로를 자동으로 탐색합니다. Flash Attention과 같은 최적화도 자동으로 발견할 수 있습니다. Rust로 작성되었으며, 추상화 계층을 피하기 위해 CUDA, Metal 등의 가속기 API에 직접 접근합니다.

대상 사용자

어떤 장치에서도 최대의 추론 속도가 필요한 개발자 및 연구자, 그리고 torch.compile를 통해 Luminal을 컴파일러 백엔드로 사용하고자 하는 PyTorch 사용자에게 적합합니다.

주요 특징

  • 고성능: H100에서 Q8 Llama 3 8B를 이론적 최대 성능의 약 80%로 실행 가능.
  • PyTorch 통합: PyTorch 모델의 컴파일러 백엔드로 작동.
  • 미니멀리스트 아키텍처: 15개의 기본 연산만을 사용하는 RISC 스타일 아키텍처.
  • 기호적 차원: 동적 형태를 기호적 차원으로 네이티브 지원하여 강력한 특화를 가능하게 함.
  • 네이티브 Rust 구현: 하드웨어 API에 직접 접근하는 정적 링크 가능한 Rust 크레이트.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트