Forward Deployed Engineer – Machine Learning (Pittsburgh)

Gray Swan AI•Pittsburgh, PA

20h•$115,439 - $178,100

About The Position

As a Forward-Deployed Machine Learning Engineer at Gray Swan AI, you’ll design, build, and deploy AI safety and security solutions directly inside customer environments. You’ll apply state-of-the-art deep learning methods to turn ambiguous, high-stakes problems into production-grade models and systems, rapidly prototyping, iterating, and hardening solutions for real-world use. If you are an engineer who doesn’t want to stick to just the same workflows, tasks, and stakeholders all the time, this role is probably perfect for you. This role blends applied research, software engineering, and customer delivery, with close collaboration across customers, sales, product, and engineering. You’ll own outcomes end-to-end and operate effectively in fast-moving, customer-facing settings.

Requirements

Hands-on experience in collaborative environments, working closely with cross-functional teams like data scientists, software engineers, and product managers.
Demonstrated ability to work independently in ambiguous environments and take full ownership of outcomes.
Hands-on experience in building and deploying machine learning models and systems.
Demonstrated expertise in designing, training, and deploying deep learning models with frameworks like PyTorch.
Demonstrated experience programming in Python and C++.
Practical experience in developing scalable machine learning pipelines and integrating them with cloud infrastructure (e.g., AWS, GCP, Azure).
In-depth knowledge of neural network architectures, such as sequence models, transformers, and other state-of-the-art approaches.
Strong algorithmic problem-solving skills and comprehensive knowledge of ML theory and optimization techniques.
Proficiency in data preprocessing and transformation, and handling large-scale datasets with multiple modalities.
Bachelor’s degree in Computer Science, Machine Learning, Engineering, or a related technical field is required.

Nice To Haves

A Master’s degree in a relevant technical field, especially with a focus on machine learning and AI safety.
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.
Strong collaboration and problem-solving abilities, with an emphasis on driving impactful solutions.
Experience working with customers, partners, or other external stakeholders to deploy and ensure the success of software products, ideally in a technical pre-sales, technical support, forward-deployed engineering, or similar role.

Responsibilities

Implement data pipelines, model serving, and evaluation frameworks.
Adapt and fine-tune models to domain-specific data and performance requirements.
Work with customer teams to understand data constraints, and operational realities.
Translate real-world problems into well-scoped proofs-of-concept and deployment architectures.
Act as the primary AI/ML technical owner for customer projects, rapidly identifying, scoping, and solving technical issues.
Develop proofs of concept quickly to validate technical feasibility and customer value.
Iterate based on user feedback, empirical results, and changing requirements.
Balance speed with long-term maintainability.
Partner with sales, product, research, and platform teams.
Communicate findings, tradeoffs, and recommendations.
Contribute learnings from deployments back into core product and platform development.