Machine Learning Researcher

Department: Machine Learning

Location: Remote (U.S. or International)

Compensation: $183K – $278K • Offers Equity • Offers Bonus

Employment Type: FullTime

About Gray Swan

Gray Swan is on a mission to empower the world to use AI safely and securely. We evaluate AI models for the leading frontier labs along with building real-time threat detection and adaptive adversarial red teaming agents for teams deploying AI.

Were a team of approximately 50 people, well-funded, growing quickly. Our work directly influences how the world deploys AI agents and systems at scale..

Learn more about how we work.

The Role

As a Machine Learning Researcher at Gray Swan AI, you will work at the boundary between research and production — developing new approaches to adversarial testing, model evaluation, and robust inference that directly inform how secure AI systems are built and deployed.

AI security is not a solved problem. This role is fundamentally about research: designing experiments, prototyping novel methods, analyzing results empirically, and translating findings into real systems that withstand adversarial pressure. You will collaborate closely with engineering and platform teams to see your ideas through from prototype to production impact.

What Youll Do

• Design and develop novel ML approaches to adversarial testing, model evaluation, and robust inference.

• Build and deploy ML models that meet real-world performance and scalability requirements.

• Design experiments, analyze results empirically, and communicate findings through publications and internal research reports.

• Develop and advance methodologies for controlling, monitoring, and analyzing ML models in production environments.

• Translate research ideas into scalable AI systems deployed in real-world, adversarial settings.

• Work closely with cross-functional teams to ensure research outcomes inform production systems.

Who You Are

Education

• Bachelors degree in Computer Science, Machine Learning, Engineering, or a related technical field.

• A Masters or PhD in a relevant technical field is strongly preferred, especially with a focus on machine learning and AI safety.

Experience

• Experience building and deploying ML models and systems.

• Demonstrated expertise in designing, training, and deploying deep learning models, particularly with PyTorch.

• Strong Python programming skills; C++ preferred.

• Experience developing scalable ML pipelines and integrating with cloud infrastructure (AWS, GCP, or Azure).

• ML research experience: building research prototype systems, designing experiments, empirical analysis of results, and communicating results via publications.

Core Technical Skills

• Neural network architectures, including transformers, sequence models, and other state-of-the-art approaches.

• Strong algorithmic problem-solving skills and knowledge of ML theory and optimization.

• Data preprocessing, transformation, and handling large-scale, multi-modal datasets.

Bonus Points If You Have

• Experience with LLMs (training, fine-tuning, or analyzing), synthetic data generation, and AI safety or security work.

• AI safety practices: model validation, robustness testing, and continuous monitoring for safety and security incidents.

• Familiarity with AI safety and security assessments and adversarial testing.

Youll Thrive Here If

• You are genuinely excited by the intersection of research and engineering, and want to both develop new AI safety ideas and see them running in real systems.

• You are motivated by real-world impact and want your work to directly influence how major AI companies deploy models right now (we work with many of the leading AI labs).

• You are eager to deepen your AI safety expertise by working alongside a team that includes some of the most respected and influential thinkers in the field.

• You thrive in a fast-paced, dynamic startup environment where ambiguity is expected.

• You bring strong collaboration and problem-solving skills, with a focus on driving meaningful, lasting impact.
  

Compensation & Benefits

We offer a competitive compensation package designed to reward impact and incentivize growth. Our compensation philosophy is informed by our current valuation and recent industry data.

Salary: $183,000-$278,000

Equity: Competitive equity package

Benefits:

• 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