Edge AI & Generative 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

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.