Machine Learning Engineer – Autonomous Perception & AI Systems

emagine; Stockholms kommun, Danmark; On-site; SEK 900,000 - 1,300,000; Full time

Job summary

emagine seeks a highly skilled Machine Learning Engineer to build next‑generation multi‑modal perception and autonomy systems for autonomous platforms. You’ll join a team advancing AI capabilities across computer vision, sensor fusion, robotics, and foundation models in mission‑critical settings.

The role combines software engineering with hands‑on ML, taking R&D concepts from prototype to deployment. You will work on embedded/edge deployments using NVIDIA Jetson, CUDA/TensorRT, and ROS2 in a

Qualifications

• 3–5+ years of experience in Machine Learning, Robotics, Computer Vision, AI, or related field.
• Strong practical experience with PyTorch and/or TensorFlow; end-to-end model development and deployment.

Responsibilities

• Design, train, and optimize ML/DL solutions for multi-modal perception, 3D object detection, scene understanding, and behavior modeling.
• Develop and maintain scalable data and training pipelines, including data curation, synthetic data, augmentation, and evaluation.
• Work with VLA models, foundation models for robotics, RL and imitation learning, and multi-modal transformer architectures.

Skills

PyTorch

TensorFlow

Python

End-to-end ML deployment

Data pipelines

GPU acceleration

C/C++

ROS2

CUDA/TensorRT

NVIDIA Jetson

Tools

NVIDIA Isaac Sim

Alpamayo

Vision-Language Models

Job description

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