Staff ML Engineer, Mid-Level — Training & Agent

Hark; San Jose (CA); On-site; USD 180,000 - 450,000; Full time

Job summary

Hark is seeking a Member of Technical Staff - Mid-Training to lead the development of training strategies for advanced AI models. This role focuses on improving agent capabilities and conducting rigorous experimentation to enhance model performance.

Ideal candidates will have a strong background in machine learning and reinforcement learning, along with experience in large model training. The position offers a competitive annual salary ranging from $180,000 to $450,000.

Qualifications

• Hands-on experience training or fine-tuning large models.
• Understanding of reinforcement learning and policy optimization.
• Experience in simulation or execution environments.

Responsibilities

• Design and implement mid-training strategies.
• Scale synthetic data generation pipelines.
• Build and optimize distributed training pipelines.
• Develop evaluation frameworks for model capability.
• Conduct experimentation to understand training dynamics.
• Collaborate cross-functionally with teams.
• Drive technical innovation in model training.

Skills

Machine learning

Reinforcement learning

Python

PyTorch

Experimentation

Job description

Hark is seeking a Member of Technical Staff - Mid-Training to lead the development of training strategies for advanced AI models. This role focuses on improving agent capabilities and conducting rigorous experimentation to enhance model performance.

Ideal candidates will have a strong background in machine learning and reinforcement learning, along with experience in large model training. The position offers a competitive annual salary ranging from $180,000 to $450,000.