[Hiring] Senior Machine Learning Engineer @Gray Swan AI

Role Description

As a Senior Machine Learning Engineer at Gray Swan AI, you will play a pivotal role in shaping the future of AI safety solutions. Research at Gray Swan AI is tightly tied to real-world impact. AI security is not a solved problem, and this role is a mix of applied research and system building:

• Developing new approaches to adversarial testing, model evaluation, and robust inference that directly inform how secure AI systems are deployed in practice.
• Working at the boundary between research and production, translating novel ideas into scalable AI systems that withstand adversarial pressure.
• Advancing methodologies for controlling, monitoring, and analyzing these models, ensuring they meet the rigorous demands of production environments.

Join us at Gray Swan AI to leverage your deep technical knowledge and leadership skills while making a significant impact in the field of AI safety!

Qualifications

• Bachelor’s degree in Computer Science, Machine Learning, Engineering, or a related technical field is required.
• A Master’s or PhD in a relevant technical field is strongly preferred, especially with a focus on machine learning and AI safety.
• 6+ years of hands-on experience in building and deploying machine learning models and systems.
• Experience with modern ML methods such as LLMs (training, finetuning, and/or analyzing), synthetic data generation pipelines, and AI safety or security work.
• Demonstrated expertise in designing, training, and deploying deep learning models with frameworks like PyTorch.
• 6+ years of experience programming in Python and C++ (preferred).
• Practical experience developing scalable machine learning pipelines and integrating them with cloud infrastructure (e.g., AWS, GCP, Azure).
• Experience conducting ML research, including building research prototype systems, experiment design, empirical analysis of results, and communicating results via publications.

Requirements

• In-depth knowledge of neural network architectures, including sequence models, transformers, and other state-of-the-art approaches.
• Strong algorithmic problem-solving skills and comprehensive knowledge of ML algorithms.
• Proficiency in data preprocessing, transformation, and handling large-scale, multi-modal datasets.

Bonus Points If You Have

• Technical leadership experience, including leading project teams or mentoring junior engineers.
• Experience with AI safety practices such as model validation, robustness testing, and continuous monitoring for safety and security incidents throughout deployment.
• Familiarity with AI safety and security assessments and adversarial testing.
• Hands-on experience working in collaborative, cross-functional environments with data scientists, software engineers, and product managers.
• Strong communication skills for articulating complex technical concepts and influencing team strategies.

Benefits

• Competitive compensation package designed to reward impact and incentivize growth.
• Salary: $194,650 - 277,400
• Equity: Competitive equity package
• 401k with up to 4% matching
• 28 days annual leave (vacation + holidays)
• Health, dental, and vision coverage
• Catered lunches (Pittsburgh office)
• Flexible work arrangements
• Visa sponsorship available for exceptional candidates