OpenAI의 오픈 웨이트 gpt-oss-120b 및 gpt-oss-20b 언어 모델과 참조 추론·도구·클라이언트 코드
fla-org/flash-linear-attention
선형 어텐션, 희소 어텐션, 상태 공간 모델 및 하이브리드 LLM 아키텍처를 위한 Triton 기반 커널과 학습용 레이어입니다.
- 스타
- 5,952
- 포크
- 778
- 열린 이슈
- 126
- 마지막 커밋
- 6시간 전
어떤 프로젝트인가요?
Flash Linear Attention(fla)은 표준 softmax 어텐션을 대체하거나 보완하는 시퀀스 모델용 하드웨어 효율적인 빌딩 블록을 제공하는 Python 라이브러리입니다. Triton 커널과 GatedDeltaNet, RetNet, GLA, Mamba2 등의 바로 사용할 수 있는 레이어, 그리고 이를 기반으로 구축된 🤗 Transformers 호환 모델 클래스를 제공합니다. 프로젝트는 구현이 플랫폼에 종속되지 않으며 NVIDIA, AMD, Intel 하드웨어에서 검증되었다고 설명합니다.
개별 레이어 외에도 fla는 게이팅을 결합한 RMSNorm, fused linear cross entropy, rotary embedding 등의 융합 모듈을 제공해 메모리 사용량을 줄이고 학습 속도를 높이는 것을 목표로 합니다. 구성 인수와 레이어별 설정을 통해 하이브리드 모델의 선형 어텐션 레이어에 표준 어텐션 레이어를 섞을 수 있습니다. 생성은 표준 Hugging Face API를 통해 수행하며, 사전 학습 체크포인트는 fla-hub 조직에서 제공합니다.
저장소에는 torchtitan을 기반으로 한 동반 프레임워크 flame을 이용한 학습 과정도 문서화되어 있으며, lm-evaluation-harness와 perplexity 측정을 사용하는 평가 스크립트도 포함되어 있습니다.
핵심 기능
- 01선형 어텐션, 희소 어텐션 및 상태 공간 모델 변형을 위한 Triton 커널.
- 02MultiScaleRetention처럼 multihead attention을 대체하는 드롭인 토큰 믹싱 레이어.
- 03구성 가능한 config를 갖춘 Hugging Face Transformers 호환 모델 클래스.
- 04원하는 깊이에 표준 어텐션 레이어를 삽입하는 하이브리드 모델.
- 05메모리 사용량을 줄이고 학습 속도를 높이는 융합 norm, rotary embedding 및 cross entropy 모듈.
- 06CUDA, ROCm, XPU, Ascend NPU 및 CPU용 백엔드 추가 패키지.
활용 사례
- 긴 컨텍스트 언어 모델에서 이차 복잡도의 어텐션을 선형 어텐션 변형으로 대체.
- Hugging Face 스타일 워크플로로 준이차 복잡도 LLM을 사전 학습하거나 파인튜닝.
- 어텐션 레이어와 순환 레이어를 혼합하는 하이브리드 아키텍처 실험.
- 여러 GPU에서 선형 어텐션 커널 속도 벤치마킹.
토픽
자주 묻는 질문
- flash-linear-attention은(는) 무엇인가요?
- 선형 어텐션, 희소 어텐션, 상태 공간 모델 및 하이브리드 LLM 아키텍처를 위한 Triton 기반 커널과 학습용 레이어입니다. Flash Linear Attention(fla)은 표준 softmax 어텐션을 대체하거나 보완하는 시퀀스 모델용 하드웨어 효율적인 빌딩 블록을 제공하는 Python 라이브러리입니다. Triton 커널과 GatedDeltaNet, RetNet, GLA, Mamba2 등의 바로 사용할 수 있는 레이어, 그리고 이를 기반으로 구축된 🤗 Transformers 호환 모델 클래스를 제공합니다. 프로젝트는 구현이 플랫폼에 종속되지 않으며 NVIDIA, AMD, Intel 하드웨어에서 검증되었다고 설명합니다.
- flash-linear-attention은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
- flash-linear-attention은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
- flash-linear-attention은(는) 아직 유지보수되고 있나요?
- 활발히 유지보수 중. flash-linear-attention의 마지막 커밋은 2026년 10월 20일입니다.
- flash-linear-attention은(는) 어디에 사용할 수 있나요?
- 긴 컨텍스트 언어 모델에서 이차 복잡도의 어텐션을 선형 어텐션 변형으로 대체. Hugging Face 스타일 워크플로로 준이차 복잡도 LLM을 사전 학습하거나 파인튜닝. 어텐션 레이어와 순환 레이어를 혼합하는 하이브리드 아키텍처 실험. 여러 GPU에서 선형 어텐션 커널 속도 벤치마킹.
- flash-linear-attention은(는) 어디서 받을 수 있나요?
- 소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/fla-org/flash-linear-attention
요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.
관련 프로젝트
선형 확산 트랜스포머를 사용해 고해상도 이미지와 동영상을 효율적으로 생성하고, 학습 및 추론 파이프라인을 제공하는 오픈소스 코드베이스입니다.
Hugging Face Hub의 저장소를 다운로드하고 업로드하며 관리하는 공식 CLI(hf) 및 Python 클라이언트입니다.
재사용 가능한 AI 에이전트 스킬을 검색·생성·평가·분석·라우팅하는 Python SDK와 CLI를 갖춘 오픈 인프라입니다.
NEO는 픽셀과 단어를 하나의 인코더 없는 dense 아키텍처로 처리하는 네이티브 VLM 제품군입니다.
FSDP2, Muon, Liger 커널을 지원하는 통합 멀티모달 모델 학습 엔진