Member of Technical Staff - Evaluation Infrastructure

SpaceXAI; Palo Alto (CA); On-site; USD 180,000 - 440,000; Full time

Benefits offered by this job

Equity

Medical, Vision, Dental

401(k) retirement plan

Disability insurance

Life insurance

Discounts and perks

Job summary

SpaceXAI, a Palo Alto-based company, is building AI systems to understand the universe. Our RL infrastructure team looks for an engineer to develop evaluation infrastructure and scale our evaluations.

We operate with a flat structure where hands-on contributions and initiative shape leadership, and clear communication is expected across the team. You will build reliable infrastructure, collaborate with modeling and evaluation teams, and tackle bottlenecks across inference, capacity management,

Qualifications

• Experience building, debugging, and optimizing efficiency of large-scale distributed systems.
• Willingness to dive deep and solve hard problems at all levels of the stack.

Responsibilities

• Build highly reliable, efficient, and easy to use infrastructure that runs all evaluations at scale.
• Collaborate closely with modeling & evaluation teams on developing new evals and tooling to support training models.
• Identify and resolve performance bottlenecks in the eval infrastructure stack, including inference and orchestration.

Skills

Distributed systems

Deep problem solving

Communication skills

Tools

LLM inference

GPU compute management

Job description

Member of Technical Staff - Evaluation Infrastructure

SpaceXAI's mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for individuals who appreciate challenging themselves and thrive on curiosity. We operate with a flat organizational structure. All employees are expected to be hands-on and to contribute directly to the company's mission. Leadership is given to those who show initiative and consistently deliver excellence. Work ethic and strong prioritization skills are important. All employees are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.

ABOUT THE ROLE:

The RL infrastructure team is looking for an engineer to help develop our evaluation infrastructure.

RESPONSIBILITIES:

• Build highly reliable, efficient, and easy to use infrastructure that runs all our evaluations at scale
• Collaborate closely with modeling & evaluation teams on developing new evals, maintaining evaluation signal quality, and building internal tooling to support training our next-generation of models
• Identify and resolve performance bottlenecks in all layers of the eval infrastructure stack, including inference, capacity fleet management, asynchronous eval orchestration, and more

BASIC QUALIFICATIONS:

• Experience in building, debugging, and optimizing efficiency of large-scale distributed systems
• Willingness to dive deep and solve hard problems at all levels of the stack

PREFERRED SKILLS AND EXPERIENCE:

• Experience in LLM inference
• Experience in GPU compute management, workload scheduling, and dynamic resource optimization
• Experience in developing interfaces for comparing models and evals

COMPENSATION AND BENEFITS:

$180,000 - $440,000 USD

Base salary is just one part of our total rewards package at SpaceXAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.