Machine Learning Engineer

Full-time
San Francisco, CA (On-site)
AI Research
$200,000 - $350,000 + equity
Posted 52 days ago

Job Description

Work on cutting-edge AI systems at one of the world's leading AI research labs. You'll help build and deploy large language models and AI applications. Key Responsibilities: • Train and fine-tune large language models • Build ML infrastructure for model training and serving • Optimize model performance and inference latency • Collaborate with researchers on new model architectures • Develop evaluation frameworks and benchmarks

Requirements

• MS/PhD in Computer Science, ML, or related field • 3+ years of production ML experience • Strong Python and PyTorch/TensorFlow skills • Experience with distributed training (DeepSpeed, FSDP) • Knowledge of transformer architectures and LLMs • Experience with GPU optimization and CUDA is a plus

Apply for this Position

Applying for: Machine Learning Engineer