Member of Technical Staff - Mid-training

23 hours, 9 minutes ago
Full-time
Lead
Software Development
x.ai

x.ai

x.ai is a technology company founded in 2014 that specializes in developing artificial intelligence personal assistants. Their flagship product is a personal assistant that efficiently schedules meetings for users. With a focus on sustainable business ...

Internet Software & Services
51-250
Founded 2023
$44M raised

Description

  • Scale synthetic coding data to trillions of tokens using large-scale Docker verification.
  • Distill knowledge from flagship models into flash models through synthetic data generation.
  • Optimize mid-training data mixtures to improve reinforcement learning performance.
  • Engineer data recipes for long-context training.
  • Develop robust and diverse evaluations for mid-training checkpoints.

Requirements

  • Expertise in machine learning and large model scaling, with familiarity across scaling laws.
  • Strong ability to design machine learning experiments.
  • Familiarity with state-of-the-art techniques for curating AI training data across text, image, audio, and video modalities.
  • Strong engineering ability in Spark, Ray, and other large-scale data processing frameworks.
  • Hands-on, high-ownership mindset in a flat organizational structure.
  • Strong communication skills and ability to share knowledge concisely and accurately.
  • Work ethic, prioritization skills, and curiosity are important.
  • Experience with large-scale Docker verification is preferred.
  • Experience working with synthetic data generation or mid-training workflows is preferred.

Benefits

  • $180,000 - $440,000 USD base salary.
  • Equity as part of the total rewards package.
  • Comprehensive medical, vision, and dental coverage.
  • Access to a 401(k) retirement plan.
  • Short-term and long-term disability insurance.
  • Life insurance.
  • Various discounts and perks.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior Research Engineer - Experimental Engineering

Consumer Reports 251-1K Media

Consumer Reports is hiring a Senior Research Engineer to lead early-stage experimental engineering efforts focused on AI, machine learning, and emerging technologies for products like AskCR.

FastAPI Machine Learning Python React React Native
23 hours, 24 minutes ago

Principal ML Scientist, Multimodal Biological Reasoning

Flagship Pioneering 251-1K Biotechnology

Flagship Pioneering’s Pioneering Intelligence is seeking a Lead to shape and advance AI-driven biological reasoning systems that support scientific discovery across the company and its portfolio.

LLM Machine Learning
1 week ago

Researcher, Evaluations

Epoch AI 1-10 Professional Services

Epoch AI is hiring a Researcher to evaluate frontier AI models on realistic, hard-to-grade tasks that reflect real office work and inform its broader AI research.

Machine Learning Python
1 week ago

Senior AI Scientist

Atria Institute 51-250 Health Care Providers & Services

Atria Health Institute is hiring a Senior AI Scientist to develop and validate medical AI models for preventive and longevity-focused care using rich longitudinal clinical and multimodal health data.

Computer Vision Hugging Face MLflow NLP Python PyTorch Reinforcement Learning Transformers
1 week, 1 day ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers