Senior AI Engineer

About the client

Our client is a VC-backed startup on a mission to transform how design teams create branded content through generative AI. They're building a world-class AI team that blends cutting-edge research with product-driven engineering to create tools that designers love.

They value autonomy, technical excellence, and bold thinking. If you're excited to push the boundaries of what's possible in generative AI while delivering tangible impact, they'd love to hear from you.

About the Role

We're looking for a Senior AI Engineer to lead the development and optimisation of generative AI features across both research and production. You'll work closely with the Head of AI to improve model quality, streamline pipelines, and launch features that reach real users. This is a high-impact role with significant ownership and autonomy.

What You'll Do

• Lead the development of new AI-powered features from concept to production
• Build and optimise training pipelines and inference systems
• Design and implement benchmarks to measure and improve model performance
• Collaborate with researchers to bring cutting-edge models into production
• Mentor junior engineers and guide architectural decisions
• Drive speed, scalability, and reliability across their AI systems

Requirements

• 5+ years of experience in production ML, with a focus on computer vision or image generation
• Expert-level proficiency in PyTorch and large-scale ML infrastructure
• Experience with diffusion models, GANs, or modern generative architectures
• Proven track record optimising models for quality, speed, and cost
• Strong quantitative and benchmarking skills
• Bachelors or Masters in CS, AI, or related field
• Must be based in Vietnam

Bonus Points

• Experience shipping AI features in creative or design tools
• Published research in top conferences (e.g., CVPR, ICCV, NeurIPS)
• Proven ability to translate academic research into production systems
• Strong systems design and architecture experience