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