Technical Lead, Machine Learning (AI)

TL;DR

Technical Lead, Machine Learning (AI): Building and scaling production-grade ML systems for a proactive AI chat application with an accent on model training, inference architecture, and data pipeline reliability. Focus on integrating large-scale models into real-world workflows, optimizing GPU performance, and ensuring system stability under heavy production constraints.

Location: Must be based in China (Beijing)

Company

hirify.global is building a proactive AI assistant designed to handle real-world tasks and long-running workflows for everyday users.

What you will do

• Architect end-to-end ML execution systems including data pipelines, training loops, and inference architecture.
• Fine-tune and adapt large models using modern techniques like LoRA, QLoRA, SFT, and DPO.
• Optimize production deployment for latency, cost, memory efficiency, and GPU utilization.
• Build evaluation pipelines to ensure model robustness, safety, and performance against real-world data.
• Collaborate with application engineering teams to integrate ML models into product surfaces.
• Lead technical execution, mentor team members, and drive pragmatic production decisions.

Requirements

• Location: Must be based in China
• Proven track record of shipping production-grade ML systems to real users.
• Expertise in Python and deep learning frameworks such as PyTorch or JAX.
• Deep understanding of large-scale model training and failure modes.
• Ability to write high-quality, production-grade code with a focus on system correctness.
• Strong communication skills and ability to thrive in small, high-trust team environments.

Hiring process

• 3 to 4 interviews with technical team members.
• Interviews conducted via virtual meetings or onsite.