Deep Learning Engineer (computer vision)

Staff Data Scientist (Computer Vision)

Company Description

Drawing on nearly four centuries of collective experience across Wabtec, GE Transportation and Faiveley Transport, the company has unmatched digital expertise, technological innovation, and world class manufacturing and services, enabling the digital rail and transit ecosystems. Wabtec is focused on performance that drives progress, creating transportation solutions that move and improve the world. Wabtec has approximately 27,000 employees in facilities throughout the world.
Visit: http://www.WabtecCorp.com

Position Overview

We are seeking a Staff level Deep Learning Engineer to design, develop, and operationalize advanced computer vision models for large scale remote visual inspection systems. This role is ideal for someone who can work independently, define technical direction, and build state of the art deep learning pipelines from the ground up.

You will be responsible for the full lifecycle of deep learning systems data preparation, model design, training, optimization, deployment, and performance monitoring while collaborating closely with MLOps engineers, software teams, and product groups. As a Staff engineer, you will act as a technical authority, mentor other engineers, and establish engineering excellence across the organization.

Remote Visual Inspection (RVI) systems enable high precision, non-contact inspection of critical industrial components using advanced imaging, optics, and AI driven analytics. In this role, you will shape the next generation of intelligent inspection capabilities by developing deep learning models that enable accurate, real-time detection, measurement, and classification in challenging, high-speed environments. You will build and optimize algorithms that support large scale inference pipelines and integrate seamlessly with cutting edge camera and sensor systems.

Responsibilities

Deep Learning & Computer Vision (Staff Level)

• Define and own the overall deep learning architecture for computer vision systems across the organization.
• Design, train, and optimize models for detection, segmentation, classification, and tracking.
• Lead technical decisions on model architectures, training strategies, evaluation methodologies, and optimization techniques.
• Ensure scalable, efficient, and robust inference for cloud, edge, and embedded deployments.
• Establish best practices for model development, data preparation, experimentation, and model lifecycle management.

Data Engineering for Deep Learning

• Define data requirements and collaborate with data teams to build high quality training and evaluation datasets.
• Develop dataset preparation, augmentation, curation, and validation pipelines.
• Contribute to the evolution of the computer vision data lake and related data infrastructure.

Model Deployment & Integration

• Collaborate closely with MLOps teams to integrate deep learning models into production grade pipelines.
• Optimize models for real-time, large-scale inference, including quantization, pruning, and hardware specific acceleration.
• Support deployment across cloud platforms (AWS, GCP, Azure) and edge environments.

Cross Team Collaboration & Leadership

• Mentor junior engineers and guide the team as the technical authority for deep learning and computer vision.
• Work closely with algorithm developers, MLOps engineers, software engineers, and product teams.
• Champion software excellence, model reliability, and performance across the organization.
• Stay current on deep learning research, emerging architectures, and industry best practices.

Requirements

Core Technical Skills

• Bachelor’s/Master’s degree in Computer Science, Engineering, or related field.
• 7+ years of experience in Deep Learning, Computer Vision, and Python (staff level contribution expected).
• Strong understanding of ML fundamentals, optimization, and model lifecycle management.
• Advanced Python skills; proficiency in C++ is a strong plus.

Deep Learning & CV

• Hands on experience with PyTorch, TensorFlow, and modern CV frameworks.
• Strong experience building and deploying production grade CV systems.
• Expertise in object detection, semantic/instance segmentation, and tracking models (e.g., YOLO, SSD, Faster R CNN, transformers).
• Experience with model optimization for latency, throughput, and memory efficiency.

MLOps & Data (Collaboration Focus)

• Familiarity with MLflow, Kubeflow, TFX, or similar platforms for model tracking and lifecycle management.
• Understanding of dataset management systems, scalable data pipelines, and feature storage.
• Experience collaborating with MLOps teams on model deployment and monitoring.

Cloud & Deployment

• Experience deploying ML models on Azure ML, AWS SageMaker, or GCP Vertex AI.
• Strong understanding of Docker, Kubernetes, and GitLab CI/GitHub Actions.
• Understanding of hardware acceleration (GPU, NPU, tensor cores) and edge inference constraints.

Soft Skills

• Demonstrated technical leadership on complex AI systems.
• Excellent problem solving, communication, and collaboration skills.
• Ability to operate independently, influence architecture decisions, and drive innovation.

What could you accomplish in a place that puts People First?

At Wabtec, it’s not just about a job - it’s about the impact you make. When our people come together, we’re Expanding the Possible by continuously improving what we do and how we do it - for our clients and each other.

If you’re ready to revolutionize how the world moves for future generations, Wabtec is the place for you.
 

Who are we?

Wabtec is a leading global provider of equipment, systems, digital solutions, and value-added services for the freight and transit rail sectors. Drawing on more than 150 years of experience, we are leading the way in safety, efficiency, reliability, innovation, and productivity. Whether it’s freight, transit, ports, logistics, mining, industrial, or marine, our expertise, technologies, and people together – are accelerating the future of transportation. With roots that date back to George Westinghouse, Thomas Edison, and Louis Faiveley, Wabtec has always built technologies and implemented solutions for a variety of sectors that are critical to meeting the needs of customers and governments alike.

Our global team of about 30,000 employees worldwide delivers performance that moves the world forward. We’re lifelong learners, obsessed with better. Learn more at www.WabtecCorp.com.

Culture powers us and the possibilities.

We believe the best ideas come from a mix of experiences and backgrounds. At Wabtec, we strive every day to create a place where everyone belongs. We’re building a culture where leadership, inclusion and your unique perspective fuel progress.

We’re proud to be an Equal Opportunity Employer. We welcome talent of all backgrounds, experiences, and identities, including race, gender, age, disability, veteran status and more.

Need accommodation? Just let us know - we’ve got you.