[2027 Campus Hire] AI Algorithm Engineer

re-zoo-me; Singapore; Hybrid; SGD 120,000 - 180,000; Full time

Job summary

re-zoo-me in Singapore seeks a passionate AI engineer to advance generative AI on mobile and embedded devices. You will design and optimize language and multi-modal models, with a focus on transformer and RNN/SSM architectures for edge deployment.

You will work with on-device frameworks such as TFLite and ExecuTorch, implement quantization and pruning, and explore RL and on-device learning to push the boundaries of on-device intelligence.

Qualifications

• Strong knowledge of transformer and LLM architectures.
• Experience with on-device AI frameworks (TFLite, ExecuTorch).
• Proficient in Python, PyTorch and C/C++.
• Understanding of deep learning model development and training methods.
• Experience deploying mobile/embedded DL algorithms is a plus.

Responsibilities

• Generative AI: language and multi-modal model algorithm development
• Transformer-based architecture development (LLaMa etc.)
• RNN and SSM-based architecture development (Mamba, Linear Attention, etc.)
• AI vision / voice / NLP algorithm development
• Edge AI: 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

Skills

Transformer models

On-device frameworks

Python

PyTorch

C/C++

Deep learning & gen AI

Tools

TFLite

ExecuTorch

Job description

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

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