Edge AI Algorithm Engineer: Transformers & On-Device DL

MediaTek Research Lab Inc.; Singapore; On-site; SGD 60,000 - 95,000; Full time

Job summary

MediaTek Research Lab Inc. in Singapore is seeking an AI Algorithm Engineer for campus hiring. You will work on Generative AI, transformer-based architectures, and edge AI algorithms for mobile and embedded devices.

The role emphasizes experience with Python, PyTorch, and C/C++, plus knowledge of NLP, vision, and speech AI to optimize models for on-device inference.

Qualifications

• Strong understanding of AI models (especially transformer and LLMs) — graph structure, operators and inference behaviour.
• Familiar with on-device frameworks for mobile/embedded devices: TFLite, ExecuTorch.
• Required programming languages: Python, PyTorch, C/C++.
• Knowledge of deep learning and generative AI algorithm/model development, optimization and training methodology preferred.
• Experience developing deep learning and generative AI algorithms for mobile/embedded devices is an advantage.

Responsibilities

• Develop generative AI algorithms for language and multi-modal models.
• Explore transformer-based architectures (LLaMa etc.).
• Investigate RNN and SSM-based architectures (Mamba, Linear Attention, etc.).
• Advance AI vision / voice / NLP algorithms.
• Optimize Edge AI algorithms for devices and edge deployment.

Skills

Transformer models

On-device ML

Python

PyTorch

C/C++

Edge AI

NLP

Vision AI

Tools

TFLite

ExecuTorch

Job description

[2027 Campus Hire] AI Algorithm Engineer

Category

Chip Design

Location

Singapore

Experience

No Work Expe.

Job Description

[What you will do]- Generative AI: language and multi-modal model algorithm development- Transformer-based architecture (LLaMa etc.)- RNN and SSM-based architecture (Mamba, Linear Attention, etc.)- AI vision / voice / NLP algorithm development- Edge AI (APU) algorithm development and optimization- Neural network optimization (quantization / pruning / low-bit techniques)- Neural network design automation- Advanced AI algorithm: reinforcement learning (RL) and on-device learning#LI-WC1

Main Requirements and Qualifications

• [Are you the right talent?]
• - Strong understanding of AI models (especially transformer and LLMs) — graph structure, operators and inference behaviour
• -Familiar with on-device frameworks for mobile/embedded devices: TFLite, ExecuTorch
• - Required programming languages: Python, PyTorch, C/C++
• - Knowledge of deep learning and generative AI algorithm/model development, optimization and training methodology preferred
• - Experience developing deep learning and generative AI algorithms for mobile/embedded devices is an advantage