AI Accelerator Architect for Embedded Systems - Stockholm - ref. d61537015

The successful candidate will have the chance to contribute to groundbreaking innovations and collaborate with leading experts in the field.

• Design and develop efficient AI/ML models and applications for deployment on mobile devices and user equipment.
• Utilize state-of-the-art deep learning architectures like GNN, ResNets, Transformers, and other neural network models, optimizing them for on-device performance.
• Integrate and optimize AI/ML models with mobile device hardware, including GPUs, specialized AI accelerators, and embedded systems.
• Analyze and evaluate the performance of AI/ML models using various datasets and metrics, iterating and refining the models as needed.

• Ph.D. degree in Computer Science, Electrical Engineering, Mathematics, Control Theory, or related fields with a strong focus on AI/ML.
• Extensive hands-on experience with AI/ML frameworks such as PyTorch, TensorFlow, MXNet, JAX, and proficiency in programming languages like Python.
• Solid understanding of deep learning architectures, neural network design, and optimization techniques for efficient model deployment.
• Familiarity with mobile device platforms, embedded systems, and hardware acceleration for AI/ML workloads, and knowledge of communication systems like 5G-NR and baseband signal processing algorithms is a strong plus.

• Excellent problem-solving skills, demonstrated ability to learn quickly and adapt to new technologies and challenges, with the ability to handle challenges, work under pressure, and deliver results in a team-based environment,
• Strong communication and collaboration skills, with the ability to work effectively in cross-functional and international teams.

Location: May also need to work at our client's location in Lund, Sweden