google-deepmind/simply
Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.
해결하는 문제
Simply는 최소한의 확장 가능한 코드베이스를 제공하여 인간과 AI 에이전트 모두가 쉽게 수정하고 반복할 수 있도록 하여 최첨단 LLM 연구를 가속화하도록 설계되었습니다. 복잡한 추상화와 무거운 의존성을 제거하여 새로운 옵티마이저, 훈련 손실 또는 RL 알고리즘과 같은 새로운 아이디어를 구현하고 테스트하는 데 필요한 시간을 줄여줍니다.
작동 방식
JAX로 구축된 이 프로젝트는 연구자가 빠르게 프로토타입을 만들 수 있는 자체 포함 환경을 제공합니다. 훈련, 평가 및 서빙(gRPC를 통한)을 위한 다양한 도구를 포함하며, 로컬 머신, Google Cloud TPU 및 XPK를 사용한 GKE에서 실험 실행을 지원합니다. 또한 장기 자율 연구를 위한 내장 에이전트 하네스와 Amplio 하네스를 갖추고 있어 AI 에이전트가 코드를 읽고, 변경 사항을 제안하고, 실험을 자율적으로 실행할 수 있습니다.
대상 사용자
LLM 연구를 위한 최소한의 코드베이스를 원하는 AI 연구자 및 개발자, 그리고 자동화된 연구 작업을 수행할 수 있는 AI 에이전트를 개발하는 사람들.
주요 특징
- 미니멀리스트 디자인: 코드를 쉽게 읽고 해킹할 수 있도록 최소한의 추상화로 JAX를 사용합니다.
- 에이전트 준비: AI 에이전트가 연구 실험을 자율적으로 반복할 수 있는 환경으로 특별히 설계되었습니다.
- 확장 가능한 인프라: 고성능 훈련을 위한 TPU 및 GKE 통합을 지원합니다.
- 소스 모델 zoo: GLM-5, Kimi K3 및 Qwen3.8과 같은 아키텍처를 위한 자체 포함 모델 플러그인을 포함합니다.
- 연구 벤치마킹: AI 에이전트의 연구 수행 능력을 벤치마킹하기 위한 11개의 점수화된 연구 과제 모음을 제공합니다.
- Amplio 하네스: 지속성과 웹 UI를 갖춘 장기 연구를 위한 Go 기반 에이전트 하네스입니다.
SUMMARY: 인간 연구자와 AI 에이전트 모두가 빠르게 반복할 수 있도록 설계된 최첨단 LLM 연구를 위한 최소한의 확장 가능한 JAX 기반 코드베이스입니다.
TITLE: LLM 개발을 자동화하고 가속화하기 위한 최소한의 JAX 기반 연구 환경
관련
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트