Member of Technical Staff - Mid-training

3 weeks, 2 days 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

PhD Physicist (AI Research & Evaluation)

Gramian Consultancy Group Professional Services

Gramian Consultancy is hiring a remote freelance PhD-level physicist for a project improving how advanced AI systems understand, analyze, and explain complex scientific topics.

1 day ago

Physics Research Collaborator (Part-time)

Weekday 11-50 Construction & Engineering

A leading AI research initiative is hiring remote, part-time Physics Research Collaborators to help evaluate and improve AI systems for advanced scientific reasoning in frontier areas of physics.

1 day ago

STEM Researcher - Computational Fields

Weekday 11-50 Construction & Engineering

A fully remote, full-time contractor role for a client developing AI evaluation benchmarks for frontier models by applying research methods from STEM and other research-intensive fields.

Git Jupyter LLM Machine Learning Python
1 day ago

Chemistry Research Collaborator (Part-time)

Weekday 11-50 Construction & Engineering

A leading AI research initiative is hiring remote Chemistry Research Collaborators to help train and evaluate AI systems on advanced scientific reasoning across frontier chemistry topics.

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