Intern, Machine Learning Engineer

Gray Swan

📍Remote (Global) 🕔 Full Time
🔄 Rolling Applications

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.