tracel-ai/models

해결하는 문제

이 저장소는 Burn 딥러닝 프레임워크를 사용하여 구현된 딥러닝 모델의 중앙 집중형 컬렉션을 제공합니다. 개발자들은 이를 통해 처음부터 구현할 필요 없이 다양한 AI 작업용 사전 구축된 모델 아키텍처에 접근할 수 있습니다.

작동 방식

이 프로젝트는 공식 및 커뮤니티 기여 모델의 세트를 정리합니다. 이러한 구현은 Burn 프레임워크 위에 구축되어 있으며, Rust 기반 딥러닝을 가능하게 합니다. 텍스트, 이미지, 오디오 등 다양한 모달리티를 포함합니다.

대상 사용자

Burn 프레임워크를 사용하고 언어 모델링, 이미지 분류, 음성 인식 등의 작업용 즉시 사용 가능한 모델 아키텍처가 필요한 개발자 및 연구자.

주요 특징

• Llama, ALBERT, ResNet 등의 공식 모델 구현.
• Stable Diffusion, Whisper, RWKV v7 등 커뮤니티 기여 모델.
• 다양한 모달리티 지원: 텍스트(LLM, 인코더), 비전(분류, 객체 탐지), 오디오(음성 인식, 음악 노트 탐지).
• MIT 및 Apache 2.0 라이선스 하에 배포.