Job Details

This range is provided by HartleyCo. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$175,000.00/yr - $250,000.00/yr

About Our Client:

Our client is assembling a world-class team to push the boundaries of AI research and development. Their mission is to build models that generalize and adapt to novel problems, bridging cutting-edge research with real-world applications.

The Role:

As a Research Engineer in Pre-Training, you'll develop and implement advanced pre-training techniques to enhance foundation models. Working closely with Research Scientists, you'll experiment with model architectures, training strategies, and large-scale data pipelines to drive AI innovation.

What You'll Do:

• Design and optimize novel pre-training methods to improve model performance.
• Conduct large-scale training experiments and analyze results.
• Collaborate with data and product teams to refine models for real-world applications.
• Build scalable pre-training pipelines for massive datasets.

What We're Looking For:

• Strong ML fundamentals and hands-on experience with large-scale models.
• Proficiency in Python, JAX, PyTorch, or TensorFlow.
• Experience with frameworks like Megatron, DeepSpeed, or MaxText.
• Problem-solving mindset with a research-driven approach.
• Ability to deploy pre-trained models in production environments.

This is a fully on-site role in San Francisco, with relocation assistance available. If you're passionate about AI research, we'd love to hear from you!

Seniority level

Mid-Senior level

Employment type

Full-time

Job function

Information Technology

Industries

IT Services and IT Consulting