Machine Learning Engineer Intern

Machine Learning Engineer Intern at Gray Swan AI In Office (Pittsburgh, PA) | Remote (International) About Gray Swan AI: Gray Swan's mission is to maximize the benefits of AI, by making it safe for everyone. As a growing startup, we develop cutting-edge security solutions for AI systems, helping organizations ensure safety and reliability in their AI-driven products. Role Overview: We are seeking a Machine Learning Engineer Intern to work alongside our research and engineering teams. This role offers hands-on experience in developing, testing, and deploying AI security solutions. You will contribute to machine learning model evaluations, build tools for AI risk assessment, and help integrate ML-driven insights into our products. Key Responsibilities: Support research on AI robustness, adversarial attacks, and defense strategies Assist in developing and testing machine learning models' security Build and optimize tools for AI risk assessment and adversarial testing Collaborate with engineers and researchers to integrate ML solutions into production environments Required Skills and Qualifications: Currently pursuing or have completed a Master's or Ph.D. in Computer Science, Machine Learning, AI, or a related field Work experience OR research involving the following: Experience in building and experimenting with machine learning models using frameworks like PyTorch either through coursework, personal projects, or internships Familiarity with Python (experience with C++ is a plus) and basic programming skills for implementing ML models Understanding of fundamental neural network architectures, such as sequence models, transformers, and other deep learning approaches Ability to preprocess, transform, and analyze datasets, with exposure to working with large or multimodal datasets Strong problem-solving skills and a basic understanding of ML theory, optimization techniques, and algorithmic principles Bonus Points: Exposure to cloud platforms like AWS, GCP, or Azure for running ML experiments or deploying models Experience with building small-scale ML pipelines or integrating ML models into applications Interest in AI security, adversarial ML, or techniques for evaluating model robustness Experience conducting empirical analysis of ML experiments, even at a research prototype level This internship provides a unique opportunity to gain hands-on experience in AI security while working with cutting-edge research and real-world applications. If you're passionate about making AI safer, Gray Swan is the right opportunity for you. This is a paid internship ranging from $30-$50/hour dependent upon experience and skills; we do work with OPT and CPT visa holders

Gray Swan's mission is to maximize the benefits of AI, by making it safe for everyone. As a growing startup, we develop cutting-edge security solutions for AI systems, helping organizations ensure safety and reliability in their AI-driven products. These organizations include OpenAI, UK AISI and other leading AI companies.