Latest AI JobsBrowse Jobs
bjakcareer logo

Lead Engineer, Machine Learning

bjakcareerChinaActAI Engineering
Apply on company site
Salary
Not listed
Job type
Full-time
Work mode
Remote
Experience
Lead
Posted
28 Sept 2026
Status
Open · checked 7d ago

About this role

Located in China, the Lead Engineer, Machine Learning position at bjakcareer centers on driving the execution layer for proactive, intelligence-backed applications. This hands-on leadership role suits an experienced machine learning professional who wants to transform research capabilities into stable, scalable production systems.

Role Overview

The Lead Engineer, Machine Learning oversees the end-to-end artificial intelligence infrastructure at bjakcareer. The position requires managing the complete model lifecycle, which encompasses data management, model training, evaluation, inference, and final deployment. The work targets reliable, long-running workflows designed to handle everyday tasks without complex prompting.

Key Responsibilities

Manage the complete machine learning systems lifecycle, covering data, training, evaluation, inference, and deployment.
Create and develop training and fine-tuning procedures for large models.
Establish evaluation frameworks to track capability, safety, robustness, and practical product performance.
Construct high-performance inference systems to optimize latency, hardware utilization, memory footprint, cost, and dependability.
Develop data pipelines for gathering both real-world and synthetic training data.
Set up dependable production infrastructure to deploy, monitor, and refine models over time.
Collaborate directly with research and application teams to convert model functions into product updates.
Execute pragmatic technical decisions and iterate quickly based on real-world outcomes.

Requirements

Demonstrated background in building and deploying production-grade machine learning systems rather than research demos.
Knowledge of modern large-model training, fine-tuning, assessment, and inference.
Solid software engineering and systems principles.
Practical background handling machine learning workloads at scale, particularly GPU-based architectures.
Sound technical judgment capable of resolving ambiguous challenges autonomously.
Preference for experimentation, quantifiable metrics, and regular shipping.
Strict standards regarding correctness, dependability, and production quality.

Skills and Tools

Python
PyTorch
JAX
GPU-based training and inference systems

About the Company

bjakcareer develops proactive applications intended to bring intelligence to conversations, errands, organization, and daily workflows with minimal prompting. The organization operates as a compact, skilled, hands-on team where engineers maintain broad ownership and make rapid decisions.

How to Apply

Interested candidates should apply on the official company careers page.

Skills and tech stack

PythonPyTorchJAXGPU-based training and inference systemsMachine LearningFine-tuning

Questions candidates ask

Where is the Lead Engineer, Machine Learning role located?

The position is located in China.

What does the role involve regarding the machine learning lifecycle?

The role covers the entire lifecycle, including data handling, model training, evaluation, inference, and deployment.

What technologies and frameworks are used in the tech stack?

The tech stack includes Python, PyTorch, JAX, and GPU-based training and inference systems.

How many interview rounds can candidates expect?

Candidates will complete between 3 and 4 interviews, conducted virtually or onsite.

What kind of team environment does bjakcareer maintain?

The organization functions as a small, high-talent-density, hands-on team where engineers maintain broad ownership and work independently.

Ready to apply?

The application is on the official bjakcareer careers page.

Apply on company site