Job Description

Role Summary/Purpose

The Machine Learning Engineer, working as part of Wabtec’s Digital Advanced Technologies (DAT) team will be responsible for developing and deploying new advanced AI solutions throughout the freight and passenger transportation industry. DAT is a cross-disciplined entrepreneurial group that carries out applied research in AI/ML, designs and deploys cutting edge algorithms and new digital solutions throughout the Wabtec portfolio of businesses. The Machine Learning Engineer will work closely with internal stakeholders to deploy scalable machine learning and deep learning solutions in cloud and edge video analytics, robotics and automation systems. 

Essential Responsibilities

• Drive the execution of design, analysis, or evaluation of scalable AI solutions at Wabtec
• Work closely with the DAT ML engineers to build and deploy highly scalable AI solutions that drive business value for our business units and customers
• Be a machine learning cross-functional engineer with organizations such as, Freight Engineering, Services, Transit Engineering, Digital Electronics, Freight Components, Mining and Marine operations to drive new computer vision solutions and opportunities 
• Participate in designing and executing data science pipelines including edge device management, data collection, exploratory data analysis, ground truth annotation, building machine/deep learning models and their deployment
• Devise and utilize algorithms and models to mine big-data stores; perform data and error analysis to improve models; clean and validate data for uniformity and accuracy
• Participate in discussions and host webinars on latest trends in artificial intelligence and digital technologies
• Develop subject matter expertise in driving digital industrial solutions across all functional engineering areas

Eligibility Requirements:

• Must be willing to work out of an office located in Bangalore, India

Qualifications / Requirements

• Master’s in engineering, Physics, Chemistry, Mathematics, Robotics or Computer Science from an accredited university or college, or relevant and significant industry experience.
• Minimum 2 years of hands-on experience in the fields of AI/ML
• Proficient in python and hand-on experience with pair programming tools like Github copilot.
• Proficient with deep learning frameworks – (Open CV, TensorFlow, PyTorch, HuggingFace, Langchain)
• Proficient in software design principles and tools such as Gitlab
• Working knowledge in Natural language processing and/or Large Language Models (LLM)
• Working knowledge in AWS model oriented services deploying and orchestrating ML/LLM workloads on Bedrock or Sagemaker services.

Desired Characteristics

• Master’s degree in Physics, Mathematics, Robotics or Computer Science from an accredited university or college, or relevant and significant industry experience
• Experience with machine learning and statistical algorithms
• Experience with developing deep learning models for computer vision applications such as object detection and segmentation
• Experience with embedded edge devices such as NVIDIA Jetson boards and video analytics applications
• Strong interpersonal, leadership and communication skills 
• A critical thinker that can quickly understand a new problem space and apply analytic techniques to identify potential value
• Self-starter with the ability to lead the effort with minimum supervision

• Ability to engage and build positive relationships with both customers and internal and external employees
• Ability to work in a cross-functional, global environment
• Effective team building and problem-solving abilities

Additional Information

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.