Optimizerai
ML Engineer
AI·머신러닝연구·R&D인프라·DevOps
AI 요약
OptimizerAI의 리서치팀 초기 멤버를 찾는 ML 엔지니어 채용공고입니다. 대규모 오디오 파운데이션 모델 구축을 위해 데이터 인프라 설계, 대규모 데이터셋 관리 도구 개발, 7B+ 파라미터 모델의 분산 학습·최적화 및 추론 속도 개선 등을 수행합니다.
주요 업무
데이터 수집 및 관리 엔지니어링 프로젝트 리드; 데이터셋을 시각화·평가·필터링하는 도구 및 모델 구축; 내부 연구자가 개발한 ML 모델을 7B+ 스케일로 확장; 분산 병렬 시스템을 위한 최적화 기법 개발; 대규모 모델의 추론 속도 최적화; 데이터 처리, 학습, 추론 및 배포용 모델·시스템의 효율적 구현.
자격 요건
Python 숙련; JAX, PyTorch 또는 TensorFlow로 머신러닝 모델 개발 경험; DDP, FSDP, Tensor Parallel 등 분산 병렬 개념에 대한 이해 및 고도로 병렬화된 PyTorch 코드 최적화 경험; Transformer, VAE, Denoising Diffusion 모델 등 기본 ML 모델을 스크래치로 구현한 경험. (우대) CUDA 커널 작성 등 저수준 최적화 경험, 대규모 데이터 관리 경험, 7B+ 모델 프리트레이닝 경험, Kubernetes 및 Docker 숙련, 멀티 노드 분산 학습 경험.
기술 스택
PythonJAXPyTorchTensorFlowDDPFSDPTensor ParallelTransformerVAEDenoising DiffusionCUDA (커널 개발)KubernetesDocker멀티노드 분산 학습