Machine Learning Engineer – Autonomous Perception & AI Systems

We are looking for a highly skilled Machine Learning Engineer with a passion for building next-generation multi-modal perception and autonomy systems. You will join a team developing advanced AI capabilities for autonomous platforms, leveraging state-of-the-art technologies in computer vision, sensor fusion, robotics, and foundation models.

The ideal candidate combines strong software engineering capabilities with hands-on machine learning expertise and has experience taking innovative R&D concepts from prototype to deployment in mission-critical environments.

Your Responsibilities

Design, train, and optimize advanced ML/DL solutions for:Multi-modal sensor fusion3D object detection and classificationScene understanding and semantic segmentationObject tracking and trajectory predictionBehavior modeling and autonomous decision supportDevelop and maintain scalable data and training pipelines, including:Data curation and labelingSynthetic data generationData augmentation and evaluationContinuous model monitoring and performance optimizationWork with cutting-edge AI architectures, including:Vision-Language-Action (VLA) modelsFoundation models for robotics and autonomyReinforcement Learning (RL) and imitation learning approachesMulti-modal transformer-based architecturesIntegrate and optimize models for deployment on embedded and edge computing platforms, utilizing technologies such as:NVIDIA GPU accelerationCUDA and TensorRTNVIDIA Jetson platformsHigh-performance inference frameworks for real-time executionEnsure accurate sensor calibration, alignment, and synchronization across complex sensor suites, mitigating drift and environmental disturbances encountered in operational environments.Collaborate closely with Systems Engineering, Planning, Controls, Robotics, and Software teams to develop safe, reliable, and deployable autonomous systems.Investigate and evaluate emerging technologies such as Alpamayo-based robotics frameworks, next-generation autonomy stacks, embodied AI, and autonomous agent architectures. We Are Looking For

3–5+ years of experience in Machine Learning, Robotics, Computer Vision, Artificial Intelligence, Applied Mathematics, Computer Science, or a related field.Strong practical experience with:PyTorch and/or TensorFlowEnd-to-end model development and deploymentTraining, evaluation, optimization, and productionization of ML systemsExperience in one or several of the following areas:Computer visionAutonomous systemsRoboticsMulti-sensor perceptionReinforcement Learning (RL)Vision-Language Models (VLM)Vision-Language-Action (VLA) systemsFamiliarity with modern AI and robotics ecosystems such as:NVIDIA Isaac Sim / Isaac LabROS2AlpamayoCUDA/TensorRTSynthetic data and simulation environmentsStrong background in:Linear algebraProbability and estimation theoryOptimizationAlgorithm designData structures and software architectureExperience developing in:PythonC/C++Ability to obtain and maintain a security clearance. Personal Qualities

We believe you are a curious and hands-on engineer who enjoys solving difficult technical challenges and continuously exploring new technologies. A strong self-driven mindset, demonstrated through personal projects within AI, robotics, autonomous systems, or software development, is highly valued.

Bonus Qualifications

Experience developing embodied AI or autonomous robotics solutions.Experience with large-scale foundation models and generative AI.Knowledge of imitation learning, world models, or agentic AI systems.Experience deploying ML workloads on NVIDIA edge hardware.Contributions to open-source robotics or machine learning projects.

Location: Stockholm or Gothenburg