Staff Machine Learning Engineer, AI Research
Job Description:
- Design, train, and evaluate machine learning models across a range of research and applied AI initiatives
- Run rapid, iterative experiments to test hypotheses and surface insights that drive model improvements
- Collaborate closely with researchers and engineers to translate cutting-edge academic advances into practical, production-ready systems
- Build and maintain robust ML pipelines for data ingestion, feature engineering, model training, and evaluation
- Optimize model performance through fine-tuning, hyperparameter search, and architecture experimentation
- Contribute to a culture of rigorous experimentation; tracking results, documenting findings, and sharing learnings with the broader team
- Stay current with the latest developments in ML and AI research, and proactively identify opportunities to apply them
- This position may require stand-by, on-call, or off-hours duties during critical research or deployment milestones
Requirements:
- Bachelor's degree in Computer Science, Mathematics, Statistics, or a related field with 5+ years of industry or research experience (Master's or PhD a plus)
- Deep hands-on experience training and evaluating ML models, including language models
- Strong proficiency in Python and ML frameworks such as PyTorch or TensorFlow
- Familiarity with MLOps tooling and infrastructure (e.g., MLflow, Weights & Biases, Kubeflow, or similar)
- Solid understanding of modern NLP, computer vision, and/or reinforcement learning techniques
- Strong ability to move fast without sacrificing rigor; you know when to prototype and when to productionize
- Excellent communication skills with the ability to clearly present experimental results to both technical and non-technical stakeholders.
Benefits:
- health, dental, vision, short-term disability, and life insurance
- paid holidays and paid time off
- fertility treatment benefit
- 401(k)
- equity
- eligibility for a discretionary company-wide bonus
Apply tot his job Apply To this Job