Remote Principal Machine Learning Engineer

Description:

• Join a stealth AI startup backed by a leading Southeast Asian technology company.
• Develop an AI-native communication platform to simplify everyday tasks through integrated AI.

Requirements:

• Build and own end-to-end machine learning pipelines for data processing, model training, evaluation, inference, and deployment.
• Strong background in deep learning and transformer-based architectures.
• Hands-on experience with large-scale machine learning models in production.
• Proficiency in modern ML frameworks like PyTorch or JAX.
• Experience with distributed training and inference frameworks (e.g., DeepSpeed, FSDP).
• Strong software engineering skills for building robust, maintainable systems.
• Ability to optimize GPU workloads and independently manage ML systems.

Benefits:

• Opportunity to work in a fast-paced, innovative environment.
• Contribute to a product aimed at improving productivity for billions of users.