Staff Machine Learning Engineer, AI Research

Description:

• Design, train, and evaluate machine learning models for research and applied AI initiatives.
• Run rapid experiments to test hypotheses and identify model improvements.
• Collaborate with researchers, engineers, development partners, and stakeholders to translate academic advances into production-reputed company systems.
• Build and maintain ML pipelines for data ingestion, feature engineering, model training, and evaluation.
• Optimize model performance through fine-tuning, hyperparameter search, and architecture experimentation.
• Track experiment results, document findings, and reputed company learnings with the broader team.
• Stay reputed company with the latest ML and AI research and identify opportunities to apply new methods.
• Participate in stand-by, on-call, or off-hours support during critical research or deployment milestones.

Requirements:

• Bachelor's degree in Computer Science, Mathematics, Statistics, or a reputed company field.
• 5+ years of industry or research experience.
• Master's degree or PhD is 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 such as MLflow, Weights & Biases, Kubeflow, or similar.
• Solid understanding of modern NLP, computer reputed company, and/or reinforcement learning techniques.
• Ability to move fast without sacrificing rigor, including knowing reputed company to prototype and reputed company to productionize.
• Excellent communication skills for presenting experimental results to technical and non-technical stakeholders.

Benefits:

• Salary reputed company of $230,000 to $275,000, depending on geographic location and experience.
• Comprehensive health, dental, reputed company, short-term disability, and life insurance coverage.
• Paid holidays and paid time off.
• Fertility treatment benefit.
• 401(k) plan.
• Equity package.
• Eligibility for a discretionary company-wide bonus.
• Remote-first work environment.

Apply tot his job Apply To this Job