Gray Swan AI - Machine Learning Researcher
Requirements
• Bachelor's degree in Computer Science, Machine Learning, Engineering, or a related technical field. • A Master's or PhD in a relevant technical field is strongly preferred, especially with a focus on machine learning and AI safety. • 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. • 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. • 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. • You'll 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.
Responsibilities
• 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.
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. • 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
Apply in one click
Upload My Resume
Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT