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 fusion

  • 3D object detection and classification

  • Scene understanding and semantic segmentation

  • Object tracking and trajectory prediction

  • Behavior modeling and autonomous decision support

• Develop and maintain scalable data and training pipelines, including:

  • Data curation and labeling

  • Synthetic data generation

  • Data augmentation and evaluation

  • Continuous model monitoring and performance optimization

• Work with cutting-edge AI architectures, including:

  • Vision-Language-Action (VLA) models

  • Foundation models for robotics and autonomy

  • Reinforcement Learning (RL) and imitation learning approaches

  • Multi-modal transformer-based architectures

• Integrate and optimize models for deployment on embedded and edge computing platforms, utilizing technologies such as:

  • NVIDIA GPU acceleration

  • CUDA and TensorRT

  • NVIDIA Jetson platforms

  • High-performance inference frameworks for real-time execution

• Ensure 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 TensorFlow

  • End-to-end model development and deployment

  • Training, evaluation, optimization, and productionization of ML systems

• Experience in one or several of the following areas:

  • Computer vision

  • Autonomous systems

  • Robotics

  • Multi-sensor perception

  • Reinforcement Learning (RL)

  • Vision-Language Models (VLM)

  • Vision-Language-Action (VLA) systems

• Familiarity with modern AI and robotics ecosystems such as:

  • NVIDIA Isaac Sim / Isaac Lab

  • ROS2

  • Alpamayo

  • CUDA/TensorRT

  • Synthetic data and simulation environments

• Strong background in:

  • Linear algebra

  • Probability and estimation theory

  • Optimization

  • Algorithm design

  • Data structures and software architecture

• Experience developing in:

  • Python

  • C/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