Machine Learning Engineer
Core
Research, train, and validate neural networks for object detection and rail detection using camera images and 3D point clouds to power autonomous rail vehicle systems.
Role type
Machine Learning Engineer (Computer Vision)
Builds
AI components for Advanced Driver Assistance Systems (ADAS) and self-driving systems for rail vehicles
Domain
Autonomous rail systems / Computer Vision
Deliverable
production ML models
Required skills
End-to-end neural network development, Python, PyTorch, ONNX, dataset management, CUDA, TensorRT
Preferred skills
Experience with multi-input neural networks, knowledge of European languages
Technologies
PyTorch, ONNX, CUDA, TensorRT
Responsibilities
Research and select neural networks for tasks like object and rail detection; Train and validate models; Create and maintain training datasets; Deploy AI models on edge devices; Supervise third-party annotation teams; Collaborate with data and robotics engineers
Seniority
Experienced IC
